Submitted by AI-without-data t3_123gy5h in deeplearning
Hi,
I'm trying to use the Open Image Dataset to train Yolov5 model. However, in the dataset, not all of object in a image are labelled. For example, there are a computer, a table, and a chair in a image, and the computer and the table are labelled with bouding-box, but the chair is not labelled. And many other images including chairs have labels of chairs.
Then, it will affect to the training. I want to know that if I want to ignore unlabelled objects in some images for computing the loss, how could I do for it?
Please let me know the solution or some websites that have the answer.
mmeeh t1_jdupe4m wrote
Either you label the chairs in the non-labelled images or you remove the chairs from the labelled images... You can try some thechniques of image manipulation to remove the non labelled chairs but unless you use photoshop or some high advanced AI to remove those chairs, it's a really bad idea to use that dataset to train a model to recognize objects that include chairs....