data format in yolo_v3.py

mystic123 / tensorflow-yolo-v3

Implementation of YOLO v3 object detector in Tensorflow (TF-Slim)

Apache License 2.0

893 stars 353 forks source link

Open violet17 opened 4 years ago

violet17 commented 4 years ago

Hi, @mystic123 Thanks for the great work.

I found it may be a bug in line 220 of yolo_v3.py:

    # transpose the inputs to NCHW
    if data_format == 'NCHW':
        inputs = tf.transpose(inputs, [0, 3, 1, 2])

IT seems that the code trys to convert NCHW to NWCH.