经过几个小时的研究,我找不到任何关于使用对象检测API的多标签预测的例子.基本上我想在图像中为每个实例预测多个标签.如下图所示:
我想预测服装类别,还有颜色和图案等属性.
根据我的理解,我需要为每个属性添加更多的分类头到第二阶段ROI特征图,并总结每个属性的损失?但是,我在对象检测代码中实现它有困难.有人可以给我一些关于我应该开始修改哪些功能的提示吗?谢谢.
tensorflow object-detection-api
object-detection-api ×1
tensorflow ×1