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

Change the outputs of the last convolutional layer to be inputs #460

Open LotfiT opened 2 years ago

LotfiT commented 2 years ago

Hi,

I would like to run some operations (sigmoid...)of a custom YOLOV4 tiny model (size 512) that are not supported by Tensil framework on CPU. To do this, I need to generate a TF lite model by changing outputs of the last convolution layers to be inputs. Any suggestions on how to do it?

Thanks.