We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent aa6f283 commit 9a2e194Copy full SHA for 9a2e194
detr_tf/data/coco.py
@@ -78,6 +78,7 @@ def get_coco_from_id(coco_id, coco, augmentation, config, img_dir):
78
image = image.astype(np.float32)
79
t_bbox = t_bbox.astype(np.float32)
80
t_class = t_class.astype(np.int64)
81
+ is_crowd = np.array(is_crowd, dtype=np.int64)
82
return image, t_bbox, t_class, is_crowd
83
84
0 commit comments