Open y78h11b09 opened 4 years ago
same here for coco datasets.
the same problem~
Changing the threshold is useless
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.
same here