AI for Quality Control: Train on Your Own Defects