安装CUDA10.1和pytorch1.5

不会,服务器自带的。

下载源码

git clone https://github.com/NVIDIA/flownet2-pytorch.git

install.sh之前

修改networks/channelnorm_package、
networks/resample2d_package、
networks/correlation_package 三个文件夹里的setup.py文件:
复现flownet2-pytorch流程
c++11改成c++14,保存。
再运行 ./install.sh(我居然过了)

运行源码前可能要改的

将datasets.py中
1、from scipy.misc import imread, imresize 替换成
from imageio import imread。
2、将main.py中的112行和125行注释掉。复现flownet2-pytorch流程

运行源码

python main.py --batch_size 8 --model FlowNet2 --optimizer=Adam --optimizer_lr=1e-4 --loss=MultiScale --loss_norm=L1 --loss_numScales=5 --loss_startScale=4 --optimizer_lr=1e-4 --crop_size 384 512 --training_dataset FlyingChairs --training_dataset_root /xxxxxxxxx/FlyingChairs/data --skip_validation

–model FlowNet2这个模型正常运行,别的好像不行
–training_dataset FlyingChairs
–skip_validation 跳过一个什么步骤

跑跑跑

复现flownet2-pytorch流程
总体记录一下,楼主没有遇到什么障碍。

相关文章:

  • 2022-12-23
  • 2021-05-21
  • 2021-11-22
  • 2022-12-23
  • 2021-10-07
  • 2022-01-27
  • 2021-09-27
猜你喜欢
  • 2022-12-23
  • 2021-12-23
  • 2021-10-22
  • 2021-05-25
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
相关资源
相似解决方案