fasterRCNN修改识别框颜色
首先进入predict.py脚本, 找到plot_img = draw_objs函数:
ctrl+左键点击draw_objs,,进入该函数:
将 colors = [ImageColor.getrgb(STANDARD_COLORS[cls % len(STANDARD_COLORS)]) for cls in classes] 这一句,修改为:
colors = [ImageColor.getrgb(STANDARD_COLORS[98]) for cls in classes]
其中: ImageColor.getrgb(STANDARD_COLORS[xx])中,xx颜色序号参照该脚本(draw_box_utils.py)开始处的STANDARD_COLORS进行修改:
需要哪个颜色就输入哪个颜色 的序号就行。比如,我要改称红色,红色为该列表中的第98个,所以用98代替xx输入就行:
colors = [ImageColor.getrgb(STANDARD_COLORS[cls % len(STANDARD_COLORS)]) for cls in classes]
然后返回predict.py运行,搞定!
ps:本文使用的fasterrcnn的代码为b站霹雳吧啦WZ的github仓库fasterrcnn代码包。