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

No Detections using Yolov4-sam-mish cfg #220

Open AdamCuellar opened 4 years ago

AdamCuellar commented 4 years ago

I'm trying to convert a yolov4-sam-mish weight file to a pb model for tensorflow. When I use detect.py I get no detections whatsoever. I've used yolov4-tiny trained on the same dataset and get detections the same detections as I would in darknet. Is the yolov4-sam-mish configuration not supported?

KuoEuran commented 2 years ago

@AdamCuellar Hi, how you converted the SAM in AlexeyAB darknet weight to tflite? Can you give me some advice? tks

AdamCuellar commented 2 years ago

@KuoEuran I was unable to get the yolov4-sam-mish working in tflite. I ended up using the compiled darknet library and darknet.py from the AlexeyAB repo. I'm sure its possible to get the SAM module in tflite but I didn't have the time to implement it myself.

KuoEuran commented 2 years ago

@AdamCuellar Thank you for your replying.