Skip to content
/ TAB Public

Official Pytorch Implementation for paper: TAB: Text-Align Anomaly Backbone Model for Industrial Inspection Tasks

License

Notifications You must be signed in to change notification settings

Howeng98/TAB

Repository files navigation

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

arXiv ViewCount

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

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}
}

About

Official Pytorch Implementation for paper: TAB: Text-Align Anomaly Backbone Model for Industrial Inspection Tasks

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages