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

文章目录

一、环境的配置

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就可以运行

相关推荐
瓢儿菜201810 天前
proteus8.17 环境配置
单片机·proteus·环境配置
mahuifa10 天前
pyqt环境配置
python·qt·pycharm·pyqt·环境配置
Roadinforest1 个月前
wsl2配置xv6全解(包括22.04Jammy)
linux·操作系统·qemu·wsl·环境配置·xv6
luoyayun3612 个月前
Trae+Qt+MSVC环境配置
vscode·qt·环境配置·trae qt
SkyXZ2 个月前
Linux下Docker及Nvidia Container ToolKit安装教程
环境配置
落日弥漫的橘_3 个月前
Node.js下载安装及环境配置教程 (详细版)
前端·node.js·环境配置·node安装教程
laugh123213 个月前
性能飞跃!TensorRT-YOLO 6.0 全面升级解析与实战指南
环境配置·tensorrt·cuda·c/c++
颜淡慕潇3 个月前
【K8S系列】在 K8S 中使用 Values 文件定制不同环境下的应用配置
云原生·容器·kubernetes·环境配置
SkyXZ3 个月前
VScode配置X11转发!让你彻底摆脱显示屏!!!
环境配置
SkyXZ3 个月前
告别虚拟机!WSL2安装配置教程!!!
环境配置