I have trained a model, named 'PCL-I2G-FF256-32frames-Modified-5e-5'. When I run the test.py with this setting below:
/public/user/anaconda3/envs/PCL-I2G/bin/python3.7 test.py --which_epoch bestval --gpu_ids 0 --partition test --average_mode after_softmax --topn 100 --force_redo --dataset_name FF-DF --real_im_path /data-x/g16/user/outputdir/original/test/ --fake_im_path /data-x/g16/user/outputdir/F2F/test/ --train_config /public/user/PCL-I2G/checkpoints_back/PCL-I2G-FF256-32frames-Modified-5e-5/opt.yml --checkpoints_dir /public/user/PCL-I2G/checkpoints_back/
an error is reported. It looks like I need a mask as an input for testing images.
Traceback (most recent call last):
File "test.py", line 257, in <module>
run_eval(opt, output_dir)
File "test.py", line 62, in run_eval
model.set_input(inputs)
File "/public/user/PCL-I2G/models/patch_inconsistency_discriminator_model.py", line 66, in set_input
self.masks = input['masks'].to(self.device)
KeyError: 'masks'
Would you tell me how to get the mask of testing images? Thanks.