Difference between revisions of "ML-TN-003 — AI at the edge: visual inspection of assembled PCBs for defect detection — Part 3"

From DAVE Developer's Wiki
Jump to: navigation, search
Line 85: Line 85:
 
===Generative adversarial networks===
 
===Generative adversarial networks===
 
===Progressive GAN implementation===
 
===Progressive GAN implementation===
 +
[[File:Training progression.png|center|thumb|500x500px|ProGAN: training progression]]
 +
 +
[[File:Fade-in progression.png|center|thumb|500x500px|ProGAN: growing progression of the model during training]]
 +
 
==Results validation==
 
==Results validation==
 
{| class="wikitable" style="text-align:center;"
 
{| class="wikitable" style="text-align:center;"

Revision as of 13:19, 14 April 2021

Info Box
NeuralNetwork.png Applies to Machine Learning


History[edit | edit source]

Version Date Notes
1.0.0 March 2021 First public release

Introduction[edit | edit source]

Building the dataset[edit | edit source]

Defects generation and acquisition[edit | edit source]

Anomalies, generation process and numerosity
Anomaly P&P Serigraphy Manual Numerosity
Missing 21
Manhattan 4
Shift x-axis 51
Shift y-axis 58
Shift&Rotation
(shift x+z-axes)
57
Rotation
(z-axis)

55
Under soldering All
Over soldering All

Class subdivision and labelling[edit | edit source]

Soldering regions extraction[edit | edit source]

Data augmentation with image synthesis[edit | edit source]

Generative adversarial networks[edit | edit source]

Progressive GAN implementation[edit | edit source]

ProGAN: training progression
ProGAN: growing progression of the model during training

Results validation[edit | edit source]

Class Synth image Resolution
(pixel)
Google Colab
(min)
AWS SageMaker
(min)

missing
full 512 × 512 ~480 ~410
upper/lower 256 × 256 ~435 ~310

tombstoning
full 512 × 512 ~460 ~390
upper/lower 256 × 256 ~420 ~300
Synthesized images for missing class
Synthesized images for tombstoning class
T-SNE algorithm results for missing class synthesized images
512 × 512 resolution full images
256 × 256 resolution full images
256 × 256 resolution upper and lower region images


T-SNE algorithm results for tombstoning class synthesized images
512 × 512 resolution full images
256 × 256 resolution full images
256 × 256 resolution upper and lower region images


Useful links[edit | edit source]