【环境配置】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就可以运行

相关推荐
SpikeKing15 天前
LLM - 使用 LLaMA-Factory 微调大模型 环境配置与训练推理 教程 (1)
人工智能·llm·大语言模型·llama·环境配置·llamafactory·训练框架
炼丹师小米2 个月前
Ubuntu24.04.1系统下VideoMamba环境配置
python·环境配置·videomamba
QH_ShareHub2 个月前
解决R包依赖版本不兼容问题
r语言·环境配置·r包
DieSnowK2 个月前
[Redis][环境配置]详细讲解
数据库·redis·分布式·缓存·环境配置·新手向·详细讲解
FORGIVEN_H2 个月前
编程环境常用命令合集
环境配置
Spy973 个月前
【最新2024】Windows系统,C++ 操作 MySQL 数据库
数据库·c++·windows·mysql·环境配置·mysql-connector
nodototao3 个月前
yolov8 save_dir设置setting文件
yolo·环境配置
SkyXZ3 个月前
Mac上HomeBrew安装及换源教程
mac·环境配置
SkyXZ3 个月前
最简最速!C++版OpenCV安装配置教程Win/Mac!!!
windows·mac·环境配置