工程链接:https://github.com/ShoufaChen/DiffusionDet
DIffusionDet需要的基础环境及各种包都配置好了,接下来我们开始用Pre-trained Model来运行demo.py
1.打包下载工程
2.将本地下载的zip文件拖进工作目录下,输入unzip [file]
解压,得到工程文件
3.下载一个训练好的模型https://github.com/ShoufaChen/DiffusionDet/releases/download/v0.1/diffdet_coco_res50.pth同样把下载好的模型拖到DiffusionDet-main目录下
4.自己随便找一张图片,名字和格式定为image.jpg,拖到DiffusionDet-main目录下
5.输入cd DiffusionDet-main
,看到demo.py
文件后运行以下命令python demo.py --config-file configs/diffdet.coco.res50.yaml \ --input image.jpg --opts MODEL.WEIGHTS diffdet_coco_res50.pth
!!!!!重点中的重点:命令其实是没有问题的,但我第二次部署在实验室的服务器上却报了以下问题
问题截图:
能看到其实图片中的预测其实是做完了的,但是到了output阶段出现了错误,查阅词条也只是知道实验室的服务器由于没装能提供用户图形界面的服务程序,但并没有解决方法……尝试了N多中方法均已失败告终
现有的资料太少,无奈只能打开作者的源码逐行试着发现问题,终于发现!!原来问题出在output
的传参这里
原因其实是作者的系统应是可以正常调起来图形界面,但我们通常的不具备这个功能,而且我们自己又没法装这个功能(我试了很多办法,没法装),所以这个代码运行不起来原因就在这。
解决思路:试着不让他在output后唤起反馈结果,而是让它的output直接输送到我自定义的一个文件夹中,问题解决!!!
解决方法:重新输入修改后的命令 python demo.py --config-file conf igs/diffdet.coco.res50.yaml --input image.jpg --output ./output/images_test.jpg --opts MODEL.WEIGHTS diffdet_coco_res50.pth
成功!