search

Wenchao-Du / LIR-for-Unsupervised-IR

This is an implementation for the CVPR2020 paper "Learning Invariant Representation for Unsupervised Image Restoration"
https://arxiv.org/pdf/2003.12769.pdf
98 stars 21 forks source link

The issue in the PascalVoc.py #7

Open xiaoachen98 opened 3 years ago

xiaoachen98 commented 3 years ago

As your attempt, you want to divide the pascalvoc dataset into 4, and this .py file is just for pascalvoc dataset. But in your code : def splitlist(folder, trainfolder1, trainfolder2, testfolder): if not os.path.exists(folder): raise Exception('input folder dose not exist ! please check it !!') if not os.path.exists(trainfolder1): os.makedirs(trainfolder1) if not os.path.exists(trainfolder2): os.makedirs(trainfolder2) if not os.path.exists(testfolder): os.makedirs(testfolder)

filelist = os.listdir(folder)
sublist = random.sample(filelist, **110000**)
sublist_a = random.sample(sublist, **55000**)
   .....

The number of pictures in the folder is 110000 and 55000 respectively, but there are only 12031 images in the pascalvoc data set. How did you do it? Or is it that you made a mistake?

chqwer2 commented 2 years ago

Hi, same problem here.

Did you find the answer?