<ol>
<li><p>在标签映射pbtxt文件中添加“color”属性。i、 e</p>
<p>项目{
名称:“/m/01g317”
编号:1
显示姓名:“人”
颜色:“粉色”
}</p></li>
<li><p>打开文件“research/object_detection/protos/string_int_label_地图.proto”. 添加以下行。在</p>
<pre><code>optional string color = 4;
</code></pre>
<p>注意分号,大括号。</p></li>
<li><p>然后需要序列化数据,因此从research运行以下命令
文件夹</p>
^{pr2}$
<p>在此之前,您必须根据您的操作系统安装protobuf。</p></li>
<li><p>更改文件object_detection/utils/lable_map的函数“convert_label_map_to_categories”的代码_实用程序.py在</p>
<p>类别=[]
已添加的\u id\u列表=[]
如果不是标签地图:
label_id_offset=1
对于范围内的类_id(max_num_classes):
追加类别({
“id”:类_id+label_id_offset,
'name':'类别{}'。格式(class_id+label_id_offset)
})
退货类别
对于标签中的项目_地图项目公司名称:</p>
<pre><code>if not 0 < item.id <= max_num_classes:
logging.info(
'Ignore item %d since it falls outside of requested '
'label range.', item.id)
continue
if use_display_name and item.HasField('display_name'):
name = item.display_name
else:
name = item.name
if use_display_name and item.HasField('color'):
color = item.color
else:
color = ''
if item.id not in list_of_ids_already_added:
list_of_ids_already_added.append(item.id)
categories.append({'id': item.id, 'name': name, 'color': color})
return categories
</code></pre></li>
</ol>
<p>5.打开“object_detection/utils/visualization”文件_实用工具.py”. 转到名为的函数
“可视化图像阵列上的方框和标签”。添加以下代码</p>
<pre><code>else:
if classes[i] in category_index.keys():
class_color = category_index[classes[i]]['color']
box_to_color_map[box] = class_color
</code></pre>
<p>在代码后面</p>
<pre><code>if agnostic_mode:
box_to_color_map[box] = 'DarkOrange'
</code></pre>