FlowNet2在ubuntu18.04上机运行实战

1、配置环境

2、下载相关模块

3、修改模型中代码

3、运行推理

python main.py --inference --model FlowNet2 --save_flow --inference_dataset ImagesFromFolder --resume ./pre_train/FlowNet2_checkpoint/FlowNet2_checkpoint.pth.tar --save ./output --inference_dataset_root ./datasets/MPI-Sintel/training

总结遇到问题

你可能感兴趣的:(python,python,pytorch)