【文件属性】：
文件名称：Pytorch-HeadTrip:具有光流和深度估计的单次深梦和序列梦（Pytorch）
文件大小：62.56MB
文件格式：ZIP
更新时间：2021-03-07 17:47:39
Python
火炬头路
工作正在进行中
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代码改编自该
【文件预览】：
Pytorch-HeadTrip-main
----spyNet.py(11KB)
----dream.py(14KB)
----requirements.txt(394B)
----LICENSE(1KB)
----depth_inference.py(1KB)
----examples()
--------dream_example.gif(62.83MB)
--------example_img()
----utils.py(9KB)
----README.md(1KB)
----configs()
--------basic_conf.yaml(1KB)
----config.py(434B)