From 5db350f44f2852d4a052e66a4e0aac8f8e6d7fea Mon Sep 17 00:00:00 2001 From: YifanXu Date: Tue, 17 Oct 2023 16:27:38 +0800 Subject: [PATCH] Update collate_batch.py --- maskrcnn_benchmark/data/collate_batch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/maskrcnn_benchmark/data/collate_batch.py b/maskrcnn_benchmark/data/collate_batch.py index 0eca2f3..1febeec 100644 --- a/maskrcnn_benchmark/data/collate_batch.py +++ b/maskrcnn_benchmark/data/collate_batch.py @@ -66,7 +66,7 @@ class BatchCollator(object): positive_map_eval = batched_pos_map.float() - return images, targets, img_ids, positive_map, positive_map_eval, greenlight_map, path # debug: delete path + return images, targets, img_ids, positive_map, positive_map_eval, greenlight_map class BBoxAugCollator(object):