使用FairMOT训练好的模型自动下载预先训练模型

问题:今天想要使用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训练好的模型自动下载预先训练模型_第1张图片
这里已经选择了模型,为什么还会下载模型呢???
然后在该路径下找到了这个链接的地址使用FairMOT训练好的模型自动下载预先训练模型_第2张图片

使用FairMOT训练好的模型自动下载预先训练模型_第3张图片
在这里执行了下载模型的代码。
在FairMOT的Github项目中看到有人文类似问题,但是没有找到好的解决办法。
然后通过该算法论文可知该算法的目标检测使用的是CenterNet网络,该网络中使用到DLA作为主干网络。CenterNet和FairMOT都是同一个大佬提出得,因此我到CenterNet项目的Issue中找到了官方回复:解答
这个是作者亲自回复的!!!!!
使用FairMOT训练好的模型自动下载预先训练模型_第4张图片
意思是直接注释掉这行代码即可。使用FairMOT训练好的模型自动下载预先训练模型_第5张图片
成功解决!!!

你可能感兴趣的:(python,深度学习,人工智能,python)