ANOMALIB第一章:安装_anomalib 本地部署
ANomalib第一章
- ANomalib第一章:本地部署ANomalib
-
- 安装anomalib
-
- 通过pip安装
- 通过源码安装
- 常见问题收录
ANomalib第一章:本地部署ANomalib
该框架是基于英特尔的openvino推理平台开发的专用于工业缺陷检测场景下的模型选择、训练和性能测试平台。目前国内使用该模型用于流水线上电路板等缺陷检测,已经有工业使用该框架并开展对应的工业缺陷检测业务。
Anomalib 是一个深度学习库,旨在收集最先进的异常检测算法,以便在公共和私有数据集上进行基准测试。Anomalib 提供了近期文献中描述的异常检测算法的几种即用型实现,以及一套便于开发和实现自定义模型的工具。该库重点关注视觉异常检测,其算法目标是检测和/或定位数据集中图像或视频中的异常。
官方项目源码:anomalib
安装anomalib
anomalib是使用python开发的,所以我们需要使用python环境去部署它。这里推荐使用conda构建一个虚拟的python环境。一共有两种方式安装该框架:通过pip安装其依赖库(推荐用于openvino在生产环境对模型进行部署时),通过github拉取开源源码(推荐学习和模型训练)
conda create --name anomalib python=3.11conda activate anomalib
通过pip安装
pip install anomalib
!!!极其不建议通过pip进行安装,因为现在的anomalib能发布的依赖库版本极低,而最新的代码已经更新到2.x版本了,所以最新的建议是直接从源码安装,下面的教程内容也是基于源码的
这将使用pyproject.toml文件中的依赖项安装Anomalib CLI。Anomalib CLI是一个用于训练、推理、基准测试和超参数优化的命令行界面。如果要将库用作Python包,可以使用以下命令安装库:
# 获取安装参数的帮助anomalib install -h# 安装完整软件包(如果是初学,担心出问题可以直接用这个)anomalib install# 安装时输出详细信息anomalib install -v# 安装核心包选项仅用于通过Torch和Lightning训练和评估模型anomalib install --option core# 仅使用OpenVINO选项进行安装。这对边缘部署很有用,因为whl较小。anomalib install --option openvino
补充一点:如果在国内下载可能会因为qiang而卡死在:Installing packages... This may take a few minutes.
可以安装urllib3然后使用魔法成功安装。
pip install urllib3==1.25.11
安装成功测试:
import anomalibprint(anomalib.__version__)
输出:1.1.0
则安装成功
通过源码安装
这个就更其他项目一样了,先git拉取源码,然后直接pip安装依赖就好
git clone https://github.com/openvinotoolkit/anomalib.gitcd anomalibpip install -e .
如果想使用Anomalib CLI,还是要和上面一样安装对应模块。
常见问题收录
问题1:我用anomalib install -v安装完成后,输入anomalib train -v,提示
To use other subcommand using anomalib install
Could not find anomalib. To use this feature, ensure that you have anomalib installed.
To use any logger install it using anomalib install -v
error: argument subcommand: invalid choice: ‘train’ (choose from ‘install’)
找不到train子命令
解决方法:
新建的环境没有packaging包,pip install packaging一下再anomalib train就可以运行了
问题2:
anomalib installImportError: cannot import name \'_encode_image\' from \'ollama._client\' (D:\\AppCache\\conda\\envs\\cv3025_tainer\\lib\\site-packages\\ollama\\_client.py)
解决方法:
pip uninstall ollamapip install ollama==0.1.6
问题3:
anomalib installModuleNotFoundError: No module named \'lightning\'
解决方法:
pip install pytorch-lightningpip install lightning
问题4
某些奇怪的代码错误,可以需要安装rust进行解决。