做汽车养护的网站线上宣传推广方案
在上一篇文章中,我在Win11上通过Docker配置了pytorch,并顺利调用了GPU。在这篇文章中,我将继续完成anomalib的配置。 anomalib是一个非常完善的异常检测框架,我希望通过它来学习经典异常检测算法,并且测试这些算法在我自己的数据集上的效果。
步骤如下:
1. 从docker Hub上拉取pytorch2.3的镜像。上一篇文章中我拉取的是1.13,但经过查看anomalib的文档发现,anomalib最新版本要求torch>2。
2. 进入到镜像中,安装git
apt-get update
apt-get install -y git
3. 克隆anomalib仓库
4. 打开vscode,连接到docker容器。
5. 运行测试代码。其实在anomalib/notebooks中提供了jupyter示例代码,但我的vscode提示不能在远程容器中运行jupyter,只能够运行py文件,所以我就创建了一个测试代码。
# Import the required modules
from anomalib.data import MVTec
from anomalib.models import Patchcore
from anomalib.engine import Engine# Initialize the datamodule, model and engine