mirrors
/
MQ-Det
mirror of https://github.com/YifanXu74/MQ-Det.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update collate_batch.py

BradyFU-patch-1
YifanXu 2023-10-17 16:27:38 +08:00 committed by GitHub
parent 6ee32ec551
commit 5db350f44f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
maskrcnn_benchmark/data/collate_batch.py
View File

@ -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):