search

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
893 stars 353 forks source link

conver_weight_pb. NCHW data_format #57

Open shenyingying opened 5 years ago

shenyingying commented 5 years ago

@mystic123 hi guys thanks for you great job,I trying use your code convert_weight_pb.py to convert my.weight to my.pb,my goal is to use the my.pb file to another format(such as snpe .dlc),but in it's tools it did't support NCHW format,how can i change your code to conver my.weight to my.pb in cpu ? thanks a lot

mystic123 commented 5 years ago

Can't you just use --data_format param?

shenyingying commented 5 years ago

ye,--data_format it didn't work for me