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
文章出处登录后可见!
已经登录?立即刷新