第一步:找到项目中的文件mmdet/core/visualization/image.py
找到第153-156行代码
if segms is not None:
color_mask = mask_colors[labels[i]]
mask = segms[i].astype(bool)
img[mask] = img[mask] * 0.5 + color_mask * 0.5
将其注释掉即可
第二步:按照下面链接中大佬博主去掉mask的方式(去掉实例分割),修改相应位置的代码
https://blog.csdn.net/CSDN_X_W/article/details/123665043
我自己没改,也成功了
第三步:再使用demo工具的时候,可以看到mask已经去掉了,但是test里没试过,有兴趣可以自己去试一下
改检测框的颜色:
demo检测时,需要在mmdet/apis/inference.py,最后两行修改
此处为可的选颜色,需要将最后的数字与第一个数位置互换,应该是跟RGB通道顺序有关