我们就以padim为例。
一、主入口:
二、Padim类。
这个Padim类就在src\anomalib\models\image\padim文件夹下。
这个Padim类的父类就是 AnomalyModule,这个父类你不能改动了,里面的内容写死了。
这个Padim类,最重要的是,做了两件事:
1、搞一个self.model
这个model是PadimModel类型的。这个model几乎就是定义了forward函数,别的没干啥。
2、定义了training_step、fit等重要函数。