mystic123 / tensorflow-yolo-v3

Implementation of YOLO v3 object detector in Tensorflow (TF-Slim)
https://medium.com/@pawekapica_31302/implementing-yolo-v3-in-tensorflow-tf-slim-c3c55ff59dbe
Apache License 2.0
890 stars 353 forks source link

Help !!! YOLOv3 !!! LeakyReLU cannot be converted to IE IR #68

Open huunhan312 opened 5 years ago

huunhan312 commented 5 years ago

I convert weights to pb format as following python convert_weights_pb.py --class_names coco.names --data_format NHWC --weights_file yolov3.weights --output_graph freeze_yolov3_model.pb However, the model optimizer cannot convert LeakyReLU to IR. Command: python mo_tf.py --input_model freeze_yolov3_model.pb --tensorflow_use_custom_operations_config extensions\front\tf\yolo_v3.json --input_shape [1,416,416,3] And I got the problem with LeakyRelu here

Model Optimizer arguments: Common parameters:

System:

My system can run examples classification, detection of squeezenet, ssd-mobilenet well with CPU and NCS2.

Anyone please tell me any clue about this problem?

Thank you.

p/s: besides, deployment_tools\model_optimizer\extensions\front\tf only has yolo_v3.json and yolo_v1_v2.json. I cannot find yolo_v3-tiny.json as tutorial at here.

anhminh3105 commented 5 years ago

Hi, I took reference from this repo and made my own implementation for use in combination with the NCSes, may it help you out with what you are doing. Link is here: https://github.com/anhminh3105/YOLOw

huunhan312 commented 5 years ago

@anhminh3105 You did a great job ! Your sharing will help me a lot !