From b89cf321f913854fe0a41bbbf683e8eb5c8a7c0c Mon Sep 17 00:00:00 2001 From: Till Raab Date: Thu, 26 Oct 2023 08:40:24 +0200 Subject: [PATCH] bf --- datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datasets.py b/datasets.py index 57e810a..9f3e457 100644 --- 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