u/Akis_agr

Hello , I am building an optical sorter for olives , that rejects the damaged and rotten ones and it lets only good olives to pass. I want the sorting to be done by an ai that classifies images that it takes of the olives on the conveyor . But i am quite new to using and training ai and i would love for anyone who has knowledge on the field to give me some advice . I would like to know what kind of models are best for this applications and maybe a few tips and tricks to make the training more effective on smaller datasets . I am currently using the resnet18 model and it works somewhat good but its not satisfactory yet . Thank you in advance for any advice