Blade6570 / icface

ICface: Interpretable and Controllable Face Reenactment Using GANs
Other
157 stars 26 forks source link

Problem in dataloader #20

Closed najmemhmdb closed 3 years ago

najmemhmdb commented 3 years ago

I've run the following command in colab:

!python /content/drive/MyDrive/MSc_Project/icface-master/test_code_released_new/test.py --dataroot ./ --model pix2pix --which_model_netG resnet_6blocks --which_direction AtoB --dataset_mode aligned --norm batch --display_id 0 --batchSize 1 --loadSize 128 --fineSize 128 --no_flip --name gpubatch_resnet --how_many 1 --ndf 256 --ngf 128 --which_ref ./new_crop/6.png --gpu_ids 0 --csv_path 00116.csv --results_dir results_video

============================= I've got the following errors:

model [Pix2PixModel] was created Traceback (most recent call last): File "/content/drive/MyDrive/MSc_Project/icface-master/test_code_released_new/test.py", line 38, in main() File "/content/drive/MyDrive/MSc_Project/icface-master/test_code_released_new/test.py", line 24, in main for i, data in enumerate(dataset): File "/content/drive/MyDrive/MSc_Project/icface-master/test_code_released_new/data/custom_dataset_data_loader.py", line 44, in iter for i, data in enumerate(self.dataloader): File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 521, in next data = self._next_data() File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 1203, in _next_data return self._process_data(data) File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 1229, in _process_data data.reraise() File "/usr/local/lib/python3.7/dist-packages/torch/_utils.py", line 425, in reraise raise self.exc_type(msg) FileNotFoundError: Caught FileNotFoundError in DataLoader worker process 0. Original Traceback (most recent call last): File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop data = fetcher.fetch(index) File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/content/drive/MyDrive/MSc_Project/icface-master/test_code_released_new/data/aligned_dataset.py", line 34, in getitem df2 = pandas.read_csv(self.csv_path) File "/usr/local/lib/python3.7/dist-packages/pandas/io/parsers.py", line 688, in read_csv return _read(filepath_or_buffer, kwds) File "/usr/local/lib/python3.7/dist-packages/pandas/io/parsers.py", line 454, in _read parser = TextFileReader(fp_or_buf, kwds) File "/usr/local/lib/python3.7/dist-packages/pandas/io/parsers.py", line 948, in init self._make_engine(self.engine) File "/usr/local/lib/python3.7/dist-packages/pandas/io/parsers.py", line 1180, in _make_engine self._engine = CParserWrapper(self.f, self.options) File "/usr/local/lib/python3.7/dist-packages/pandas/io/parsers.py", line 2010, in init self._reader = parsers.TextReader(src, **kwds) File "pandas/_libs/parsers.pyx", line 382, in pandas._libs.parsers.TextReader.cinit File "pandas/_libs/parsers.pyx", line 674, in pandas._libs.parsers.TextReader._setup_parser_source FileNotFoundError: [Errno 2] No such file or directory: '00116.csv'

but, when I check it, there is 00116.csv in "/content/drive/MyDrive/MSc_Project/icface-master/test_code_released_new" directory. Could you pleas help me?

thank you.