├── .gitignore ├── README.md ├── config.yaml ├── configs ├── configDBFAD.yaml ├── configEAD.yaml ├── configMixed.yaml ├── configRD.yaml ├── configRN.yaml ├── configSN.yaml └── configSTPM.yaml ├── images ├── DistillBased.png ├── DualModel.png ├── EAD.png ├── RD.png ├── RememberingNormality.png ├── SSPCAB.png ├── STPM.png └── SingleNet.png ├── models ├── DBFAD │ ├── reverseResidual.py │ ├── trainer_dbfad.py │ └── utilsModel.py ├── EfficientAD │ ├── TeacherWeights │ │ ├── teacher_medium.pth │ │ └── teacher_small.pth │ ├── efficientAD.py │ └── trainer_ead.py ├── RememberingNormality │ ├── RD │ │ ├── bn.py │ │ ├── de_resnetRM.py │ │ └── trainer_rnrd.py │ ├── ST │ │ ├── resnetRM.py │ │ └── trainer_rnst.py │ └── memoryModule.py ├── ReverseDistillation │ ├── bn.py │ ├── de_resnet.py │ └── trainer_rd.py ├── SingleNet │ ├── ffc.py │ ├── fourierFilter.py │ ├── single_net.py │ └── trainer_sn.py ├── StudentTeacher │ ├── student.py │ ├── trainer_mixed.py │ └── trainer_st.py ├── sspcab.py ├── teacher.py ├── trainer_base.py └── utilsResnet.py ├── requirements.txt ├── train.py └── utils ├── functions.py ├── mvtec.py └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/README.md -------------------------------------------------------------------------------- /config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/config.yaml -------------------------------------------------------------------------------- /configs/configDBFAD.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/configs/configDBFAD.yaml -------------------------------------------------------------------------------- /configs/configEAD.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/configs/configEAD.yaml -------------------------------------------------------------------------------- /configs/configMixed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/configs/configMixed.yaml -------------------------------------------------------------------------------- /configs/configRD.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/configs/configRD.yaml -------------------------------------------------------------------------------- /configs/configRN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/configs/configRN.yaml -------------------------------------------------------------------------------- /configs/configSN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/configs/configSN.yaml -------------------------------------------------------------------------------- /configs/configSTPM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/configs/configSTPM.yaml -------------------------------------------------------------------------------- /images/DistillBased.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/images/DistillBased.png -------------------------------------------------------------------------------- /images/DualModel.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/images/DualModel.png -------------------------------------------------------------------------------- /images/EAD.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/images/EAD.png -------------------------------------------------------------------------------- /images/RD.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/images/RD.png -------------------------------------------------------------------------------- /images/RememberingNormality.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/images/RememberingNormality.png -------------------------------------------------------------------------------- /images/SSPCAB.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/images/SSPCAB.png -------------------------------------------------------------------------------- /images/STPM.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/images/STPM.png -------------------------------------------------------------------------------- /images/SingleNet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/images/SingleNet.png -------------------------------------------------------------------------------- /models/DBFAD/reverseResidual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/DBFAD/reverseResidual.py -------------------------------------------------------------------------------- /models/DBFAD/trainer_dbfad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/DBFAD/trainer_dbfad.py -------------------------------------------------------------------------------- /models/DBFAD/utilsModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/DBFAD/utilsModel.py -------------------------------------------------------------------------------- /models/EfficientAD/TeacherWeights/teacher_medium.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/EfficientAD/TeacherWeights/teacher_medium.pth -------------------------------------------------------------------------------- /models/EfficientAD/TeacherWeights/teacher_small.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/EfficientAD/TeacherWeights/teacher_small.pth -------------------------------------------------------------------------------- /models/EfficientAD/efficientAD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/EfficientAD/efficientAD.py -------------------------------------------------------------------------------- /models/EfficientAD/trainer_ead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/EfficientAD/trainer_ead.py -------------------------------------------------------------------------------- /models/RememberingNormality/RD/bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/RememberingNormality/RD/bn.py -------------------------------------------------------------------------------- /models/RememberingNormality/RD/de_resnetRM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/RememberingNormality/RD/de_resnetRM.py -------------------------------------------------------------------------------- /models/RememberingNormality/RD/trainer_rnrd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/RememberingNormality/RD/trainer_rnrd.py -------------------------------------------------------------------------------- /models/RememberingNormality/ST/resnetRM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/RememberingNormality/ST/resnetRM.py -------------------------------------------------------------------------------- /models/RememberingNormality/ST/trainer_rnst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/RememberingNormality/ST/trainer_rnst.py -------------------------------------------------------------------------------- /models/RememberingNormality/memoryModule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/RememberingNormality/memoryModule.py -------------------------------------------------------------------------------- /models/ReverseDistillation/bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/ReverseDistillation/bn.py -------------------------------------------------------------------------------- /models/ReverseDistillation/de_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/ReverseDistillation/de_resnet.py -------------------------------------------------------------------------------- /models/ReverseDistillation/trainer_rd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/ReverseDistillation/trainer_rd.py -------------------------------------------------------------------------------- /models/SingleNet/ffc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/SingleNet/ffc.py -------------------------------------------------------------------------------- /models/SingleNet/fourierFilter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/SingleNet/fourierFilter.py -------------------------------------------------------------------------------- /models/SingleNet/single_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/SingleNet/single_net.py -------------------------------------------------------------------------------- /models/SingleNet/trainer_sn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/SingleNet/trainer_sn.py -------------------------------------------------------------------------------- /models/StudentTeacher/student.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/StudentTeacher/student.py -------------------------------------------------------------------------------- /models/StudentTeacher/trainer_mixed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/StudentTeacher/trainer_mixed.py -------------------------------------------------------------------------------- /models/StudentTeacher/trainer_st.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/StudentTeacher/trainer_st.py -------------------------------------------------------------------------------- /models/sspcab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/sspcab.py -------------------------------------------------------------------------------- /models/teacher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/teacher.py -------------------------------------------------------------------------------- /models/trainer_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/trainer_base.py -------------------------------------------------------------------------------- /models/utilsResnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/models/utilsResnet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/train.py -------------------------------------------------------------------------------- /utils/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/utils/functions.py -------------------------------------------------------------------------------- /utils/mvtec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/utils/mvtec.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SimonThomine/DistillationAD/HEAD/utils/util.py --------------------------------------------------------------------------------