手动分割
这项工作是出于实践目的而完成的。
网络
在Pytorch中自定义实施 。 仅用于一个输出类。 用于最后提交的模型存储在MODEL.pth文件中。 它已通过上提供的模型进行了预培训,但该模型最初用于汽车细分。 在来自GTEA_gaze数据集的一部分图像上,模型的得分为0.90。
pytorch版本:0.3.1
要回滚pytorch,您只能使用
“ pip卸载火炬”
“ pip install火炬== 0.3.1”
用法
测试
尝试“ python test.py -h”以查看更多选项。
只需测试一张图像并通过运行即可重命名输出
“ python test.py -i test1.jpg -o ouput1.jpg”
通过运行测试多个图像
“ python test.py -i test1.jpg test2.jpg --model MODEL.pth”
通过运行在绘图中
2024-01-21 15:03:42
2.81MB
Python
1