problem with "data_dir" at starts...

[jdmelendez]

I have download all images, and the directory created is "PetImages" with two subdirectorys called "Cat" and "Dog".

When I run this command: ..." datasets.ImageFolder(data_dir + '/train',..." , appear an error because don't found the path.

Can you help me, please?

Thanks.

[sshadr]

The data has to be arranged this way
root/train/cats/cat0.jpg and so on (all your cat images)
root/train/dogs/ dog0.jpg and so on (all your dog images)

The same has to be done for test as well
check the documentation for torchvision.datasets.ImageFolder

[amitrajitbose]

Thanks for the answer @Adarshd19
Did that resolve your issue @Melendez95 ?