Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Handling negative images or images with no object of interest #839

Open
PrajwalCogniac opened this issue Dec 9, 2023 · 0 comments
Open

Comments

@PrajwalCogniac
Copy link

PrajwalCogniac commented Dec 9, 2023

The images I work with are damage to industrial parts with a very specific image, for that reason a defective part and another one that is fine are very similar.

The problem of taking samples randomly, is that although it is true that many parts of the background are taken there are practically no good samples of parts without defects. What ends up happening is that it relates the defective being with the piece itself (since it has not seen enough).

That is why I want to give feedback to the system so that it sees completely clean images, without objects.

I will try to make the necessary changes in the MultiBox_Loss, I think I should only add the check if the target is empty and avoid the match. Line 105

And then add checks on those elements that depend on the anchors (localization and mask). Probably return a 0 directly?

Originally posted by @adriaciurana in #137 (comment)
@adriaciurana I am facing the exact issue, I am doing on industrial images as well. Can you please let me know how did you handle this ?

Any help on this issue is useful @adriaciurana @dbolya @abhigoku10

@PrajwalCogniac PrajwalCogniac changed the title > The issue is that the whole pipeline (not just the loss computation) assumes that the image has at least one annotation. In fact, the dataset (data/coco.py) won't even consider the image if it doesn't have any annotations. That's why I recommended to add a dummy annotation instead, so there would be at least one annotation in each image that you can just discard later. Handling negative images or images with no object of interest Dec 9, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant