BiseNetV2网络复现

1.代码地址

https://github.com/CoinCheung/BiSeNet[0]

2.参考文章

https://arxiv.org/pdf/2004.02147v1.pdf[0]

【工程测试与训练】使用BiSeNetv2测试、训练cityscapes数据集、训练自己的数据集_magic_ll的博客-CSDN博客_bisenet训练自己的数据集[0]

3.遇到的问题

使用 【bisenetv2_city】测试图片:

将example图片放到tools文件夹中(否则会出现错误TypeError: ‘NoneType’ object is not subscriptable)

运行bisenetv2_city.py
打开tools/demo.py ,做如下修改:

parse.add_argument('--config', dest='config', type=str, default='../configs/bisenetv2_city.py',)
parse.add_argument('--weight-path', type=str, default='../MODEL/model_final_v2_city.pth',)
parse.add_argument('--img-path', dest='img_path', type=str, default='example.png',)

运行demo.py
结果保存为res.jpg

使用【bisenetv2_coco】测试视频:

把video视频放到tools文件夹中

运行bisenetv2_coco.py

打开tools/demo_video.py,做如下修改:

parse.add_argument('--config', dest='config', type=str, default='../configs/bisenetv2_coco.py',)
parse.add_argument('--weight-path', type=str, default='../MODEL/model_final_v2_coco.pth',)
parse.add_argument('--input', dest='input', type=str, default='video.mp4',)
parse.add_argument('--output', dest='output', type=str, default='res.mp4',)

运行demo_video.py

结果保存为res.mp4

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
扎眼的阳光的头像扎眼的阳光普通用户
上一篇 2022年5月7日
下一篇 2022年5月7日

相关推荐