【环境配置】AST: Asymmetric Student-Teacher Networks for Industrial Anomaly Detection

坚果仙人2024-09-26 13:22

文章目录

一、环境的配置
二、预处理
三、训练
四、问题

一、环境的配置

bash 复制代码

# zheP04_cmp_AST
$ conda create -n P04_cmp_AST python=3.9
$ conda activate P04_cmp_AST
$ conda install -y anaconda::scikit-learn
$ conda install -y conda-forge::scipy
$ conda install -y conda-forge::numpy

# 根据自己的cuda版本来安装相应的pytorch
$ conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
$ conda install -y conda-forge::tqdm
$ conda install -y conda-forge::efficientnet-pytorch
$ conda install -y conda-forge::tifffile
$ conda install -y conda-forge::pillow

# 我用anaconda装不上matplotlib，所以用了pip
$ pip install -y matplotlib

二、预处理

修改源代码中config.py文件的第5行和第6行（注意这两个目录不要有包含关系，要不然程序会出错）

三、训练

运行train_teacher.py文件
运行train_student.py文件
运行eval.py文件

四、问题

（1）问题一：ValueError: Last value of index pointer should be less than the size of index and data arrays

这是应为scipy的版本太低，按照我的测试，scipy 1.13就可以运行