search

david8862 / keras-YOLOv3-model-set

end-to-end YOLOv4/v3/v2 object detection pipeline, implemented on tf.keras with different technologies
MIT License
638 stars 220 forks source link

DIoU-NMS #30

Open kz258852 opened 4 years ago

kz258852 commented 4 years ago

Where do I need to modify to use the DIOU-NMS? thank you.

david8862 commented 4 years ago

Where do I need to modify to use the DIOU-NMS? thank you.

For DIoU-NMS postprocess we only have numpy/C++ implementation, you can change corresponding param option for nms_boxes to enable the numpy part during training & evaluation.