介绍
再现代码
结果
内容
src:源代码,包括模型,数据读取器和实用程序
工具:用于运行测试或可视化的主要功能
脚本:用于运行测试或可视化的脚本
其他目录是不言自明的
要求
Python2.7
OpenCV
pytorch v0.3.0
联合会
麻木
指示
预训练模型可。 将模型包括在./checkpoints目录中,或在./scripts/test.sh修改变量CHECKPOINT 。
跑步
./scripts/init_dir.sh
制作必要的目录。
跑步
./scripts/test.sh
在media目录中的图像上测试模型。 或者,您可以更改./scripts/test.sh的变量IMAGES_DIR以在您自己的图像上进行测试。
以相同的方式运行
./scripts/visualize.sh
可视化结果。 渲染的图像将保存在./results/imgs
输出格式
采用J
2024-03-20 12:49:48
998KB
Python
1