# WPFormer

**Repository Path**: ctpactwangke/WPFormer

## Basic Information

- **Project Name**: WPFormer
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-10-28
- **Last Updated**: 2025-10-28

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# [\[CVPR2025\]Wavelet and Prototype Augmented Query-based Transformer for Pixel-level Surface Defect Detection](https://openaccess.thecvf.com/content/CVPR2025/papers/Yan_Wavelet_and_Prototype_Augmented_Query-based_Transformer_for_Pixel-level_Surface_Defect_CVPR_2025_paper.pdf)

![image](.//network.png)
# Code
* The code is available at [Baidu Disk]( https://pan.baidu.com/s/1oc3gSLK7KsE4satHqBKDNQ?pwd=x7vk)(提取码: x7vk) and [Google Drive](https://drive.google.com/drive/folders/1HDRzqXynM4jp66FSBVcXoyJOrTB_jkLX?usp=sharing) 
* The saved models can be downloaded at [Google Drive](https://drive.google.com/drive/folders/1HDRzqXynM4jp66FSBVcXoyJOrTB_jkLX?usp=sharing).
# Requirements
* python==3.7.13 
* cudatoolkit==11.3.1 
* pytorch==1.11.0 
# Citation
```bibtex
@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](https://github.com/facebookresearch/MaskFormer), [Mask2Former](https://github.com/facebookresearch/Mask2Former), whose code has been utilized in this repository.