在mmrotate中如果想要修改标注框的线条粗细并且不显示标签的类别及得分进行如下两处修改:
1、mmrotate/models/detectors/base.py中修改第26行的thickness,以修改标注框的线条粗细
def show_result(self,
img,
result,
score_thr=0.3,
bbox_color=(72, 101, 241),
text_color=(72, 101, 241),
mask_color=None,
thickness=9, #默认为2,可以修改为自己需要的数值
font_size=13,
win_name='',
show=False,
wait_time=0,
out_file=None,
**kwargs):
2、mmrotate/core/visualization/image.py中将draw_lables注释掉(180至189行),即可在得到的检测结果中不显示标注结果及其得分
if bboxes is not None:
num_bboxes = bboxes.shape[0]
bbox_palette = palette_val(get_palette(bbox_color, max_label + 1))
colors = [bbox_palette[label] for label in labels[:num_bboxes]]
draw_rbboxes(ax, bboxes, colors, alpha=0.8, thickness=thickness)
horizontal_alignment = 'left'
positions = bboxes[:, :2].astype(np.int32) + thickness
areas = bboxes[:, 2] * bboxes[:, 3]
scales = _get_adaptive_scales(areas)
scores = bboxes[:, 5] if bboxes.shape[1] == 6 else None
将
# draw_labels(
# ax,
# labels[:num_bboxes],
# positions,
# scores=scores,
# class_names=class_names,
# color=text_colors,
# font_size=font_size,
# scales=scales,
# horizontal_alignment=horizontal_alignment)
修改后结果