Skip to content

Commit 74a68e3

Browse files
authored
Merge pull request #238 from ChipsSpectre/master
fix crash of video in case of no detected label.
2 parents 35f96cc + 9528da0 commit 74a68e3

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)