Hliníkové profily majú štyri bežné chyby, a to škrabance, netesnosti, preliačiny a vypuklý prášok. Súbor údajov o obrázku hliníkového profilu v tomto článku pochádza od hliníkovej spoločnosti v provincii Jiangsu. Pôvodný súbor údajov o snímkach hliníkových profilov obsahuje celkovo 342 snímok defektov hliníkového profilu. Existuje málo vzoriek obrazu defektov a podiel niektorých defektov v celom súbore údajov je príliš malý a údaje defektov sú nevyvážené. Keď je trénované hlboké učenie, ak je súbor údajov malý, spôsobí to, že model bude mať problémy s nadmerným prispôsobením.
Aby sa vyriešili vyššie uvedené problémy, tento článok používa rozšírenie obrazu na rozšírenie súboru údajov o chybných obrazoch hliníkových profilov. Technológia zväčšenia obrazu vykonáva sériu náhodných úprav kontrastu, rotácií atď. na pôvodných obrazových údajoch, aby sa vygenerovali podobné, ale odlišné tréningové údaje, aby sa zväčšila veľkosť súboru tréningových obrázkov, pričom sa zníži závislosť modelu od určitých funkcií a zlepší sa všeobecnosť modelu. . izačná schopnosť.