Error on running demo

[aditya-droid]

Hi,
Firsty, I would like to thank you for the awesome work you have been contributing towards open-source for car-model detection.
I have been getting the following error while running the demo.py file:

Also, which model are you using for the this repo?

[pikapikazhou]

I met the problem too, and if you can tell me the solution if you solve this trouble . Thank you so much

[aditya-droid]

I still haven't found the solution. I am waiting for the developer's response.

[pikapikazhou]

I still haven't found the solution. I am waiting for the developer's response.

I solved the problem yesterday.
demo.py line24:
os.path.isfile(os.path.join(test_path, f)) and f.endswith('.jpg')] #change the .jpg to .png

that's because in the images folder , the number of jpg pictures is less than 20

hope it can help you

[foamliu]

Thanks @pikapikazhou , could you please create a PR to fix this?
@aditya-droid Sorry, I've switched to PyTorch 1 year ago and don't have a keras/tf environement.