fasterRCNN修改识别框颜色

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代码包。