[CVPR2025]Wavelet and Prototype Augmented Query-based Transformer for Pixel-level Surface Defect Detection

Code

• The code is available at Baidu Disk(提取码: x7vk) and Google Drive
• The saved models can be downloaded at Google Drive.

Requirements

• python==3.7.13
• cudatoolkit==11.3.1
• pytorch==1.11.0

Citation

@inproceedings{Yan_2025_CVPR,
  title     = {Wavelet and Prototype Augmented Query-based Transformer for Pixel-level Surface Defect Detection},
  author    = {Feng Yan and Xiaoheng Jiang and Yang Lu and Jiale Cao and Dong Chen and Mingliang Xu},
  booktitle = {Proceedings of the Computer Vision and Pattern Recognition Conference (CVPR)},
  month     = {June},
  year      = {2025},
  pages     = {23860--23869},
}

Acknowledgement

We would like to acknowledge the contributions of public projects, such as MaskFormer, Mask2Former, whose code has been utilized in this repository.