hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite
https://github.com/hunglc007/tensorflow-yolov4-tflite
MIT License
2.25k stars 1.24k forks source link

Are you planning to implement CIoU? #110

Open AlexeyAB opened 4 years ago

AlexeyAB commented 4 years ago

@hunglc007 Hi, Are you planning to implement CIoU?

Examples:

Desctiption: https://medium.com/@jonathan_hui/yolov4-c9901eaa8e61

1_egnHdTEZcYJKkcPkDTgaVw

hunglc007 commented 4 years ago

Thank you. Im preparing push a newer version with improvements, which include CIoU loss and yolov4-tiny

zcdliuwei commented 4 years ago

Thank you. Im preparing push a newer version with improvements, which include CIoU loss and yolov4-tiny

the loss is nan, has this problem been solved ?

hunglc007 commented 4 years ago

Thank you. Im preparing push a newer version with improvements, which include CIoU loss and yolov4-tiny

the loss is nan, has this problem been solved ?

I updated mish activation to fix nan loss issue.

zcdliuwei commented 4 years ago

Thank you. Im preparing push a newer version with improvements, which include CIoU loss and yolov4-tiny

the loss is nan, has this problem been solved ?

I updated mish activation to fix nan loss issue.

great !!