Skip to content

Commit 9a2e194

Browse files
committed
is_crowd to int64
1 parent aa6f283 commit 9a2e194

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

detr_tf/data/coco.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,7 @@ def get_coco_from_id(coco_id, coco, augmentation, config, img_dir):
7878
image = image.astype(np.float32)
7979
t_bbox = t_bbox.astype(np.float32)
8080
t_class = t_class.astype(np.int64)
81+
is_crowd = np.array(is_crowd, dtype=np.int64)
8182
return image, t_bbox, t_class, is_crowd
8283

8384

0 commit comments

Comments
 (0)