NVIDIA / vid2vid

Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.
Other
8.62k stars 1.2k forks source link

Very Simple Problem #122

Open giorgio-borghetto opened 5 years ago

giorgio-borghetto commented 5 years ago

Goodmorning. I've a very simple issue regarding your project. I'm following the instructions, but, when I try the Testing phase with Faces, I obtain this error:

(venv) C:\Users\giorg\PycharmProjects\Progetto_Tesi\vid2vid>python test.py --name edge2face_512 --dataroot datasets/face/ --dataset_mode face --input_nc 15 --loadSize 512 --use_single_G ------------ Options ------------- add_face_disc: False aspect_ratio: 1.0 basic_point_only: False batchSize: 1 checkpoints_dir: ./checkpoints dataroot: datasets/face/ dataset_mode: face debug: False densepose_only: False display_id: 0 display_winsize: 512 feat_num: 3 fg: False fg_labels: [26] fineSize: 512 fp16: False gpu_ids: [0] how_many: 300 input_nc: 15 isTrain: False label_feat: False label_nc: 0 loadSize: 512 load_features: False load_pretrain: local_rank: 0 max_dataset_size: inf model: vid2vid nThreads: 2 n_blocks: 9 n_blocks_local: 3 n_downsample_E: 3 n_downsample_G: 3 n_frames_G: 3 n_gpus_gen: 1 n_local_enhancers: 1 n_scales_spatial: 1 name: edge2face_512 ndf: 64 nef: 32 netE: simple netG: composite ngf: 128 no_canny_edge: False no_dist_map: False no_first_img: False no_flip: False no_flow: False norm: batch ntest: inf openpose_only: False output_nc: 3 phase: test random_drop_prob: 0.05 random_scale_points: False remove_face_labels: False resize_or_crop: scaleWidth results_dir: ./results/ serial_batches: False start_frame: 0 tf_log: False use_instance: False use_real_img: False use_single_G: True which_epoch: latest -------------- End ---------------- CustomDatasetDataLoader Traceback (most recent call last): File "test.py", line 23, in data_loader = CreateDataLoader(opt) File "C:\Users\giorg\PycharmProjects\Progetto_Tesi\vid2vid\data\data_loader.py", line 6, in CreateDataLoader data_loader.initialize(opt) File "C:\Users\giorg\PycharmProjects\Progetto_Tesi\vid2vid\data\custom_dataset_data_loader.py", line 33, in initialize self.dataset = CreateDataset(opt) File "C:\Users\giorg\PycharmProjects\Progetto_Tesi\vid2vid\data\custom_dataset_data_loader.py", line 11, in CreateDataset from data.face_dataset import FaceDataset File "C:\Users\giorg\PycharmProjects\Progetto_Tesi\vid2vid\data\face_dataset.py", line 2, in import torchvision.transforms as transforms File "C:\Users\giorg\PycharmProjects\Progetto_Tesi\venv\lib\site-packages\torchvision__init.py", line 1, in from torchvision import models File "C:\Users\giorg\PycharmProjects\Progetto_Tesi\venv\lib\site-packages\torchvision\models__init__.py", line 11, in from . import detection File "C:\Users\giorg\PycharmProjects\Progetto_Tesi\venv\lib\site-packages\torchvision\models\detection\init.py", line 1, in from .faster_rcnn import * File "C:\Users\giorg\PycharmProjects\Progetto_Tesi\venv\lib\site-packages\torchvision\models\detection\faster_rcnn.py", line 7, in from torchvision.ops import misc as misc_nn_ops File "C:\Users\giorg\PycharmProjects\Progetto_Tesi\venv\lib\site-packages\torchvision\ops\init__.py", line 1, in from .boxes import nms, box_iou File "C:\Users\giorg\PycharmProjects\Progetto_Tesi\venv\lib\site-packages\torchvision\ops\boxes.py", line 2, in from torchvision import _C ImportError: DLL load failed: Impossibile trovare il modulo specificato.

Every previous steps work fine, this not and I've tried evrey possible path. Someone can help me?