Changing The Loss Function

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.23k stars 1.24k forks source link

Changing The Loss Function #481

Open bluwski opened 1 year ago

bluwski commented 1 year ago

Hey @hunglc007 , I was impressed with your work, I'd like to ask... I saw in the utils.py there are multiple losses (iou, giou, and ciou), how to tryout each losses? Like what script in a function should I replace to use a particular loss?

Thanks