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

convert yolov3.weight to yolov3.pb in --data_format NHWC #58

Open shenyingying opened 5 years ago

shenyingying commented 5 years ago

@mystic123 HI GUYS: when i use you convert_weight_pb.py to convert my modle yolov3-tiny.weight to yolov3-tiny.pb in --data_format NHWC ,it throw this bug control_input_ops) File "/home/sy/.local/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 1629, in _create_c_op raise ValueError(str(e)) ValueError: Negative dimension size caused by subtracting 2 from 1 for 'detector/yolo-v3-tiny/pool2_1/MaxPool' (op: 'MaxPool') with input shapes: [?,1,208,32]. can anyone help me,thanks a lot

mystic123 commented 5 years ago

It looks like input placeholder size is too small

shenyingying commented 5 years ago

thanks for you replay,how should i change the code?