TAB: Text-Align Anomaly Backbone Model for Industrial Inspection Tasks

Installation:

git clone [email protected]:Howeng98/TAB.git
cd TAB
conda create -n TAB
conda activate TAB
pip install -r requirements.txt

Pre-training

You can kindly modify the pre-training dataset, backbone structure, and hyperparameters in the config.py

python main.py

Anomaly Backbone Checkpoint Download

Description	Weights Download
MVTecAD ResNet18	Download Link
IndustrialDataset ResNet18	Download Link

Industrial Dataset

We propose a well-organized dataset for industrial visual inspection named Industrial Dataset. This dataset was constructed by selectively aggregating open-source datasets from global websites and official public conference workshops as references. The Industrial Dataset encompasses 30 categories, including 15 object categories and 15 texture categories, amounting to a total of 17,393 images. Notably, the Industrial dataset exclusively comprises normal images.

Dataset Structure

  Industrial Dataset
  ├─ Washer
  │  ├─ 0001.png
  │  ├─ 0002.png
  │  ...
  │  └─ 3241.png
  │  
  ├─ Metal Tube
  │  ├─ 001.png
  │  ├─ 002.png
  │  ...
  │  └─ 122.png
  ├─ Hazelnut
  ...

Dataset Sources

Idx	Class Name	Number of normal samples	Source	Note
1	Bent Metal Stripe	185	link	MPDD
2	Board	40	link	DAGM
3	Bolt	128	link	MPDD
4	Bottle	746	link
5	Brick Tile	80	link	UIUC
6	Capsule	749	link
7	Chip	1211	link	PCB-AoI Public
8	Cork	216	link	KTH-TIPS2
9	Encapsulates Carpet	728	link	-
10	Hazelnut	50	link	ADFI
11	Hazelnut Toy	34	link
12	Jagged Carpet	80	link	Kylberg
13	Leather	600	link	Leather Defect Cls
14	Metal Flat	54	link	MPDD
15	Metal ornament	289	link	MPDD
16	Metal Pole	110	link	MPDD
17	Metal Tube	122	link	MPDD
18	Plaid	20	link	UIUC
19	Rhombus Carpet	80	link	Kylberg
20	Screw	59	link	-
21	Solar Board	2624	link	ELPV
22	Sponge	40	link	KTH-TIPS2
23	Streak Board	1041	link	DAGM
24	Stripe Carpet	336	link	Kylberg
25	Stripe Wood	1000	link	DAGM
26	Tile	230	link	-
27	Toy Wheel	3200	link	-
28	Upholstery	20	link	UIUC
29	Washer	3241	link
30	Wave Carpet	80	link	Kylberg

NTHU CVLAB

For more details, please visit CVLAB.

License

MIT Licence

References

• MPDD
• Capsule
• PCB-AoI Public
• Alcohol Bottle
• Hazelnut
• Hazelnut Toy
• ELPV
• DAGM
• Kylberg
• KTH-TIPS2
• UIUC
• Leather Defect Classification

Citation

If you found TAB useful in your research or applications, please kindly cite using the following BibTeX:

@article{lee2023tab,
  title={TAB: Text-Align Anomaly Backbone Model for Industrial Inspection Tasks},
  author={Lee, Ho-Weng and Lai, Shang-Hong},
  journal={arXiv preprint arXiv:2312.09480},
  year={2023}
}