环境配置
之前的文档中记录了如何进行环境配置
Swin-Transformer-Object-Detection环境配置
图片检测
下载权重文件
将该权重文件放入wights文件夹中
运行代码
python demo/image_demo.py demo/demo.jpg configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py weights/mask_rcnn_swin_tiny_patch4_window7.pth
视频检测
python demo/video_demo.py demo/demo.mp4 configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py weights/mask_rcnn_swin_tiny_patch4_window7.pth --out video/test01.mp4
注意!video文件夹需要自己创建