size_index is missing for preprocess_test function

[aelnouby]

When trying the demo, I found that preprocess_test has two arguments (data, size_index) while in the demo.py file, only one argument is passed yolo_utils.preprocess_test((image, None, cfg.inp_size))[0] , how to fix this ?

Thanks.

[lucheng07082221]

I got the same problem

[CompilerBian]

Try

im_data = np.expand_dims(
        yolo_utils.preprocess_test((image, None, cfg.inp_size), 0)[0], 0)

Then comment this line in preprocess_test

#inp_size = inp_size[size_index]

[lucheng07082221]

@CompilerBian thank you very much ,this problem had solved

[nijkah]

@CompilerBian I did it.(editing demo.py and utils/yolo.py) But I get this error.

TypeError: 'numpy.int64' object is not iterable

Same error occurs when removing comment sign on the line you mentioned

[longcw]

Sorry for introduced this bug in the last PR.
I just fixed this in the master branch: dfdd822