BiseNetV2网络复现

1.代码地址

https://github.com/CoinCheung/BiSeNet

2.参考文章

https://arxiv.org/pdf/2004.02147v1.pdf

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

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