Convert the weights of YOLO v3 object detector into tensorflow lite format. It can be served for tensorflow serving as well.
docker build -t tflite .
docker run -it -v /home/peace195/tensorflow-lite-yolo-v3:/root/ tflite
wget https://raw.githubusercontent.com/pjreddie/darknet/master/data/coco.names
wget https://pjreddie.com/media/files/yolov3.weights
wget https://pjreddie.com/media/files/yolov3-tiny.weights
wget https://pjreddie.com/media/files/yolov3-spp.weights
python ./convert_weights_pb.py
(this can be used for tensorflow serving) tflite_convert --saved_model_dir=saved_model/ --output_file yolo_v3.tflite --saved_model_signature_key='predict'
Optional Flags
convert_weights_pb.py:
--class_names
Path to the class names file
--weights_file
Path to the desired weights file
--data_format
`NCHW` (gpu only) or `NHWC`
--tiny
Use yolov3-tiny
--spp
Use yolov3-spp
--output_graph
Location to write the output .pb graph
Contact me if you have any issues: binhtd.hust@gmail.com / Binh Do