search

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

save_model for yolov4-tiny_3l.cfg #264

Open ghost opened 3 years ago

ghost commented 3 years ago

Hi, Could you please help me to know what changes are to be made while converting the yolov4-tiny_3l.cfg in the save_model.py In my understanding as the anchors in yolov4.cfg and yolov4-tiny_3l.cfg are same, but convolutional layers vary in yolo layers. Thus I assume that YOLOv4_tiny_3l is to be written accordingly. Could you please help me to write this function. Thank you.

vinorth-v commented 3 years ago

Hi, did you manage to solve it?

ghost commented 3 years ago

@vinorth05 unfortunately not. I am still looking for the solution. However, I am currently using yolov4-tiny.cfg.

FUZhanhong commented 3 years ago

@re-shubhamturai hi, dose yolov4-tiny work well? I've changed tiny's anchors in config, but the result not good. can you give me some adivce, thank you?

ghost commented 3 years ago

@FUZhanhong, yes, yolov4-tiny is working well. I presume there are no changes required in case of yolov4-tiny. Regarding changes in yolov4-tiny: I would suggest you could look into darknet repository issues. However, my issue is regarding yolov4-tiny_3l.cfg. I am sure @hunglc007 can put some light on it.

vinorth-v commented 3 years ago

It works with these changes: https://github.com/hunglc007/tensorflow-yolov4-tflite/issues/261