Pytorch HeadTrip:具有光流和深度估计的单次深梦和序列梦(Pytorch) 源码

火炬头路 工作正在进行中 Pytorch中具有光流和深度估计的单次深梦和序列梦。 特征: 使用光流进行序列梦(Farneback或SpyNet) 深度估计 支持多种Pytorch架构 梦想中的ImageNet单班,查看我的帖子 安装 要求 Python 3.7 火炬1.7 OpenCV Matplotlib pip install -r requirements.txt 根据您的cuda版本,在环境中安装pytorch可能会出错。 如果要使用fp16,则需要手动安装 用法 为了进行推断,您需要一个配置文件,例如_configs文件夹中的基本conf 。 python dream.py --config configs/basic_conf.yaml 当前,只能梦想从视频中预先用ffmpeg提取的帧序列 SpyNet代码改编自该