bf

2023-10-26 08:40:24 +02:00
parent b1e5b9e665
commit b89cf321f9
1 changed files with 1 additions and 1 deletions
--- a/datasets.py
+++ b/datasets.py
@@ -59,7 +59,7 @@ class CustomDataset(Dataset):
        target["iscrowd"] = iscrowd
        image_id = torch.tensor([idx])
        target["image_id"] = image_id
-        target["image_name"] = image_name
+        # target["image_name"] = image_name #ToDo: implement this as 3rd return...

        return img_tensor, target