Submitted by AutoModerator t3_11pgj86 in MachineLearning
Sonicxc t1_jca1rh1 wrote
How can i train a model so that it detects severity of damage in a image. Which algo will suit for my need?
LeN3rd t1_jcgp44s wrote
How big is your dataset? Before you start anything wild, i would look at kernel clustering methods. Or even clustering without kernels. Just cluster your broken and non broken images and calculate some distance (can be done with kernels if it needs to be nonlinear).
Also Nearest neighbor could work pretty well in your case. Just compare your new image to the closest (according to some metric) in your two datasets and bobs your uncle.
If you need a number, look at simple CNNs. you need more training data though for this to work well.
Sonicxc t1_jcleo4j wrote
Hey man, thanks for the input. I will look into what you have mentioned
josejo9423 t1_jco1xea wrote
Maybe trying image classification? CNN pytorch
Viewing a single comment thread. View all comments