使用预训练模型运行DiffusionDetection

文章描述了在部署DiffusionDet项目时遇到的问题,即在没有图形界面的实验室服务器上运行预训练模型的demo.py时,在output阶段出现错误。作者发现问题在于代码依赖于图形界面,而服务器不支持。为解决这个问题,作者修改了命令,使输出直接保存到指定文件夹,避免了调用图形界面,从而成功运行了代码。
摘要由CSDN通过智能技术生成

工程链接: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

在这里插入图片描述

在这里插入图片描述
成功!

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值