标签: conv-neural-network object-detection yolo
我正在研究DeepFashion数据集上的模型,其中包括两个类(上层布料和下层布料)。数据集的问题在于,每个图像仅使用一个边界框。但是,在某些情况下,图像包括两个类。
我仍然不确定在预测阶段,使用这种数据集的训练后的模型是否能够为图像返回两个边界框,包括上衣和下衣。有想法吗?
答案 0 :(得分:1)
不用担心,您可以首先在此处尝试一些常见对象的示例-
https://github.com/tensorflow/models/blob/master/research/object_detection/object_detection_tutorial.ipynb