EfficientDet.Pytorch icon indicating copy to clipboard operation
EfficientDet.Pytorch copied to clipboard

The demo.py has too many boxes to incorrect, and the code updated is unstable.

Open y78h11b09 opened this issue 5 years ago • 5 comments

y78h11b09 avatar Jan 07 '20 11:01 y78h11b09

same here

inFreedom92 avatar Jan 08 '20 05:01 inFreedom92

same here for coco datasets.

alen-mask avatar Jan 09 '20 03:01 alen-mask

the same problem~

ma3252788 avatar Jan 09 '20 15:01 ma3252788

Changing the threshold is useless

ma3252788 avatar Jan 09 '20 16:01 ma3252788

The problem is that no filtering is performed in demo.py. After appending boxes, labels and scores, add this:

dataframe = list(zip(bboxes, labels, bbox_scores))
dataframe = list(filter(lambda x: x[2] >= args.threshold*100, dataframe))

if(len(dataframe) > 0):
    bboxes, labels, bbox_scores = list(zip(*dataframe))
else:
    bboxes, labels, bbox_scores = [], [], []

Best wishes.

xafarranxera avatar Jan 23 '20 16:01 xafarranxera