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

时间:2024-03-27 11:34:19
【文件属性】:

文件名称:Pytorch-HeadTrip:具有光流和深度估计的单次深梦和序列梦(Pytorch)

文件大小:62.56MB

文件格式:ZIP

更新时间:2024-03-27 11:34:19

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)

网友评论