重要错误:没有名为'models.flownet2_Pythorch型号'

2024-10-03 21:32:00 发布

您现在位置:Python中文网/ 问答频道 /正文

我不熟悉DL&使用Nvidia服务器 尝试运行vid2vid和 已经按照以下说明操作: https://github.com/NVIDIA/vid2vid

我好像有错误models.flownet2_Pythorch型号' 我跟踪了这里的安装: https://github.com/NVIDIA/flownet2-pytorch/blob/master/README.md#convert-official-caffe-pre-trained-models-to-pytorch

但是我得到了错误 ------------选项--------

add_face_disc: False
aspect_ratio: 1.0
basic_point_only: False
batchSize: 1
checkpoints_dir: ./checkpoints
dataroot: datasets/Cityscapes/
dataset_mode: temporal
debug: False
densepose_only: False
display_id: 0
display_winsize: 512
feat_num: 3
fg: True
fg_labels: [26]
fineSize: 512
gpu_ids: [0]
how_many: 300
input_nc: 3
isTrain: False
label_feat: False
label_nc: 35
loadSize: 2048
load_features: False
load_pretrain: 
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: 3
name: label2city_2048
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: True
use_real_img: False
use_single_G: True
which_epoch: latest
-------------- End ----------------

CustomDatasetDataLoader
dataset [TestDataset] was created
vid2vid
Traceback (most recent call last):
  File "test.py", line 25, in <module>
    model = create_model(opt)
  File "/workspace/AI/vid2vid/models/models.py", line 7, in create_model
    from .vid2vid_model_G import Vid2VidModelG
  File "/workspace/AI/vid2vid/models/vid2vid_model_G.py", line 13, in <module>
    from . import networks
  File "/workspace/AI/vid2vid/models/networks.py", line 12, in <module>
    from .flownet2_pytorch.networks.resample2d_package.resample2d import Resample2d
**ImportError: No module named 'models.flownet2_pytorch'**

我的服务器上安装的软件包是

^{pr2}$

有什么想法吗?在


Tags: noinpyfalsetrueonlymodelmodels