项目作者: srymurphy

项目描述 :
darknet to caffe model
高级语言: Python
项目地址: git://github.com/srymurphy/Model-Transformation.git
创建时间: 2020-06-10T03:00:28Z
项目社区:https://github.com/srymurphy/Model-Transformation

开源协议:

下载


Model-Transformation

darknet to caffe model

python darknet2caffe.py tiny-yolo-voc.cfg tiny-yolo-voc.weights tiny-yolo-voc.prototxt tiny-yolo-voc.caffemodel

darknet to TensorFlow

Optional Flags

convert_weights:

  1. --class_names
  2. Path to the class names file
  3. --weights_file
  4. Path to the desired weights file
  5. --data_format
  6. NCHW (gpu only) or NHWC
  7. --tiny
  8. Use yolov3-tiny
  9. --spp
  10. Use yolov3-spp
  11. --ckpt_file
  12. Output checkpoint file

convert_weights_pb.py:

  1. --class_names 1. Path to the class names file
  2. --weights_file
  3. Path to the desired weights file
  4. --data_format
  5. NCHW (gpu only) or NHWC
  6. --tiny
  7. Use yolov3-tiny
  8. --spp
  9. Use yolov3-spp
  10. --output_graph
  11. Location to write the output .pb graph to

YOLOv3 转ckpt文件格式:
python convert_weights.py —class_names coco.names —data_format NHWC —weights_file yolov3.weights

YOLOv3-tiny 转ckpt文件格式:
python convert_weights.py —class_names coco.names —data_format NHWC —weights_file yolov3-tiny.weights —tiny

YOLOv3 转pb文件格式:
python convert_weights_pb.py —class_names coco.names —data_format NHWC —weights_file yolov3.weights

YOLOv3-tiny 转pb文件格式:
python convert_weights_pb.py —class_names coco.names —data_format NHWC —weights_file yolov3-tiny.weights —tiny