0
$\begingroup$

I want to train a model with python over the images, and these images are for a metal product. my aim is to detect the defects, to notice if a product is a failure.

what kind of architecture do you suggest? should I train over the class? or should I use an autoencoder?

$\endgroup$
1

1 Answer 1

1
$\begingroup$

It sounds like you only have "normal" examples with which to train your model, so this makes the problem feel like an application for outlier detection algorithms. There are a variety of approaches here. You could indeed take an autoencoder approach and then use the reconstruction error to determine if a new image is normal or not, on the presumption that normal images will have lower error. You could also take the activations from the bottleneck layer and build an explicit outlier model using an algorithm like isolation forest. And you are not limited to just autoencoders if you take the latter approach -- other models pretrained for other tasks like image net classification could also provide good features for outlier detection.

$\endgroup$

You must log in to answer this question.

Not the answer you're looking for? Browse other questions tagged .