search

hizhangp / yolo_tensorflow

Tensorflow implementation of YOLO, including training and test phase.
MIT License
794 stars 443 forks source link

can you tell me what's the meaning of these code in test.py? #59

Open wzxyks opened 6 years ago

wzxyks commented 6 years ago

test.py： classes_num_filtered = np.argmax(filter_mat_probs, axis=3)[filter_mat_boxes[0], filter_mat_boxes[1], filter_mat_boxes[2]] why not: classes_num_filtered = np.argmax(probs, axis=3)[filter_mat_boxes[0], filter_mat_boxes[1], filter_mat_boxes[2]] ？？？ I can't understand, Please help, thank you.

Polaris-SDU commented 6 years ago

When we do prediction,we should do: step1: Setting some lower probability prediction ( class-specific confidence score for each bounding box<0.2) to zero step2: Sorting from big to small according to confidence score and then calculate IOU

so by the step1: classes_num_filtered = np.argmax(filter_mat_probs, axis=3)[filter_mat_boxes[0], filter_mat_boxes[1], filter_mat_boxes[2]] (I am not a English speaker,it maybe not clear)