yolov7 实例分割记录

1.下载mask分支源码以及yolov7-mask.pt权重文件

git clone https://github.com/WongKinYiu/yolov7.git -b mask yolov7-mask

2.安装detectron2和jupyter

git clone https://github.com/facebookresearch/detectron2
cd detectron2
python setup.py install
pip install jupyter

3.分割代码在tools/instance.ipynb,转换成.py文件

jupytejupyter nbconvert --to python tools/instance.ipynb

4.下载的是mask分支代码,但是导包报错

 只能用源码mask分支下的general.py替换,出现函数方法找不到错误同解

5.运行instance.pyyolov7 实例分割记录_第1张图片

 图片路径错误

6.yolov7 实例分割记录_第2张图片

点击upsampling.py,将

替换为

7.结果没有输出

 

你可能感兴趣的:(github)