Skip to content

Commit 9528da0

Browse files
committed
fix crash of video in case of no detected label.
1 parent 35f96cc commit 9528da0

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

efficientdet_test_videos.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@
5757
def display(preds, imgs):
5858
for i in range(len(imgs)):
5959
if len(preds[i]['rois']) == 0:
60-
continue
60+
return imgs[i]
6161

6262
for j in range(len(preds[i]['rois'])):
6363
(x1, y1, x2, y2) = preds[i]['rois'][j].astype(np.int)

0 commit comments

Comments
 (0)