Update lightning_wheat.py

Author: tappyness1

src/lightning_classes/lightning_wheat.py

@@ -83,8 +83,10 @@ def training_step(self, batch, batch_idx):
 
     def validation_step(self, batch, batch_idx):
         images, targets, image_ids = batch
-        targets = [{k: v for k, v in t.items()} for t in targets]
+        images = list(img for img in images)
+#         targets = [{k: v for k, v in t.items()} for t in targets]
         outputs = self.model(images, targets)
+        outputs = [{k: v for k, v in t.items()} for t in outputs]
         res = {target['image_id'].item(): output for target, output in zip(targets, outputs)}
         self.coco_evaluator.update(res)