nistarlwc / keras-yolo3-convert2pb

convert qqwweee/keras-yolo3 h5 file to tensorflow pb file
11 stars 3 forks source link

keras-yolo3-convert2pb

convert qqwweee/keras-yolo3 to pb file

python convert.py yolov3-tiny.cfg yolov3-tiny.weights yolov3-tiny.h5

python keras2pb.py --input_model="yolov3-tiny.h5" --output_model="yolov3-tiny.pb" --is_tiny=True --num_class=80