问题:今天想要使用FairMOT训练好的模型进行人的多目标跟踪。已经下载了fairmot_dla34.pth但是运行demo.py还是会自动下载dla34-ba72cf86.pth模型。
环境:
ubuntu,gtx 1050ti , python3.6 ,CUDA10.0
具体问题:
在输入完执行命令之后,显示:
Createing model……
Downloading: “http://dl.yf.io/dla/models/imagenet/dla34-ba72cf86.pth”
如果下的很快的话可能这里我也就忍了,但是超级超级慢。如图所示:
这里已经选择了模型,为什么还会下载模型呢???
然后在该路径下找到了这个链接的地址
在这里执行了下载模型的代码。
在FairMOT的Github项目中看到有人文类似问题,但是没有找到好的解决办法。
然后通过该算法论文可知该算法的目标检测使用的是CenterNet网络,该网络中使用到DLA作为主干网络。CenterNet和FairMOT都是同一个大佬提出得,因此我到CenterNet项目的Issue中找到了官方回复:解答
这个是作者亲自回复的!!!!!
意思是直接注释掉这行代码即可。
成功解决!!!