复现flownet2-pytorch流程

安装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文件：

c++11改成c++14，保存。
再运行 ./install.sh（我居然过了）

运行源码前可能要改的

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

运行源码

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 跳过一个什么步骤

跑跑跑

总体记录一下，楼主没有遇到什么障碍。