Има четири често срещани дефекта в алуминиевите профили, а именно драскотини, течове, вдлъбнатини и изпъкнал прах. Наборът от данни за изображения на алуминиев профил в тази статия идва от компания за алуминий в провинция Дзянсу. Оригиналният набор от данни за изображения на алуминиев профил включва общо 342 изображения на дефекти на алуминиев профил. Има малко проби от изображения на дефекти и делът на някои дефекти в целия набор от данни е твърде малък и данните за дефектите са небалансирани. Когато се обучава задълбочено обучение, ако наборът от данни е малък, това ще доведе до проблеми с пренастройването на модела.
За да разреши горните проблеми, тази статия използва увеличаване на изображението, за да разшири набора от данни за изображения на дефектни алуминиеви профили. Технологията за увеличаване на изображението извършва поредица от произволни корекции на контраста, завъртания и т.н. върху оригиналните данни на изображението, за да генерира подобни, но различни данни за обучение, за да разшири размера на набора от изображения за обучение, като същевременно намалява зависимостта на модела от определени характеристики и подобрява обобщеността на модела . способност за изация.