首先非常感谢这篇文章的帮助,让我成功跑通了测试代码
https://blog.csdn.net/weixin_46428351/article/details/115515904
1,我的设备环境
电脑系统,windows10
运行平台,pycharm
框架,pytorch, torchvision== 0.11.3 torch ==1.10.2
2, 文献 :Real-Time Seamless Single Shot 6D Object Pose Prediction
直接在bing里搜这个名字就能搜到文章
开源代码链接:https://github.com/Microsoft/singleshotpose/
3,需要的文件:
(1)去上面的开源代码链接里下载代码,
点击绿色的code,然后, Download ZIP
然后就下载好啦!, 再解压一下:
这样就可以了,
(2)然后我们再来下载作者训练好的模型和数据集,
这个可以在(1)中下载的代码文件里的 ’README.md‘文件里有链接,用记事本打开就能看到:
看到图里的四个链接,复制到浏览器打开,就能下载啦,(我的电脑要挂梯子才能下)
下载好了之后就是这样:
就是这四个文件。解压这四个文件,把他们放在什么地方呢?不要急我后面说,先解压了准备好就行。
4 现在东西都准备好了,开始整。
(1)创建一个pycharm的 project ,然后把 3 (1)的代码文件无脑的放进projiect文件里。
(2)把 3(2)保存的文件,放到project 里:
也就是这几个文件:
放进project 里, 就像这样:
这三个文件里都在这里了。
(3)在pycharm打开该项目(project)后,打开底下的 terminal
输入:
python valid.py --datacfg cfg/ape.data --modelcfg cfg/yolo-pose.cfg --weightfile backup/ape/model_backup.weights
能跑通:
(4)单个图像的可视化,看到 在project 文件下有一个,valid.ipynb文件,这个文件要用jupyter notebook打开,我是打开之后把第一行注释了其他内容复制到pycharm的新建python文件:single_obj_testing.py,中
然后同样在terminal 中运行:
python single_obj_testing.py --datacfg cfg/ape.data --modelcfg cfg/yolo-pose.cfg --weightfile backup/ape/model_backup.weights
就能出图。
(5)如果要测试其他的图像,就对应的更改代码:
如:要运行cat 的代码:
(a) single_obj_testing.py中修改为:
terminal 中的命令修改为:
python single_obj_testing.py --datacfg cfg/cat.data --modelcfg cfg/yolo-pose.cfg --weightfile backup/ape/model_backup.weights
(6)多个图像的测试,按照4(4)的方法,复制 multi_obj_pose_estimation文件夹下的 valid_multi.ipynb 来做一个新的 sssstestsss.py文件,同样要注释掉第一行。
!!!!需要注意的是 会报错:'…/xxxx’文件找不到,要修改一下这里:
就是注释的这两个地方,加上[1:]就行了,实际上是对字符串从新取了一下。
然后运行:
python sssstestsss.py cfg/yolo-pose-multi.cfg backup_multi/model_backup.weights
另外就是报错找不到某些文件,这些文件都在,multi_obj_pose_estimation文件夹下的,复制到外面来就能运行了。
希望能帮到需要的朋友啦! 下面一行有很多图标啦,随手点个赞呀!点个关注最好啦!五角星是收藏啦,说不定今天不用以后会用到啦!