运行CenterNet的demo.py,出现错误 “torch.utils.ffi is deprecated”

使用最新版本的pytorch,(torch版本1.3.1,torchvision版本0.4.2。)

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

从 https://github.com/xingyizhou/CenterNet 运行 demo.py,

出现 “ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.” 的错误。

 

解决方法:

用 https://github.com/CharlesShang/DCNv2 替换 CenterNet/src/lib/models/networks/DCNv2 。

发布了53 篇原创文章 · 获赞 12 · 访问量 13万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章