0
$\begingroup$

I use EfficientNetB0 for performing image classification with one of the class as "stone countertop table", the other class is "not stone countertop table". I use the reddest part in gradCAM to put a point saying that this is the "stone countertop table". But in some cases, it is pointing to the basin (as most of the stone countertop accompany with a basin, and lead efficientNETB0 focus on the wrong thing). Is there a way to tell EfficientNetB0 not to use basin as a hint to look for "stone countertop table", and therefore the reddest part in gradCAM will always point to the surface of "stone countertop table"?

This is what is desired:

enter image description here

This is not:

enter image description here

This is not as well:

enter image description here

$\endgroup$

1 Answer 1

0
$\begingroup$

From these 2 pictures alone, I feel that the algorithm is focusing on the table color. Your second table is black but the basin is silver. Can you add more black table images?

$\endgroup$
1
  • $\begingroup$ Thanks, I add one more image for your reference with white table + basin, it still focuses on the basin. $\endgroup$
    – william007
    Commented Jan 18, 2021 at 16:14

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge you have read our privacy policy.

Not the answer you're looking for? Browse other questions tagged or ask your own question.