🍉 博主微信 cvxiayixiao 其他专栏点击头像查看: 🍓 【Segment Anything Model】计算机视觉检测分割任务专栏。 🍑 【公开数据集预处理】特别是医疗公开数据集的接受和预处理,提供代码讲解。 🍈 【opencv+图像处理】opencv代码库讲解,结合图像处理知识,不仅仅是调库。 文章目录 1️⃣MVTec数据集和预处理 2️⃣微调框架 1. 编写Dataset.py 2. 编写test原始sam的test_ori_sam.py的文件 3. 编写train加入lora模块的train_lora.py的文件 4. 编写测试/验证文件test_lora_sam.py文件 3️⃣代码 1. Dataset.py代码 2. test_ori_sam.py代码 3. train_lora.py代码 这里我们讲一下怎么注入lora的 代码 4. test_lora_sam.py代码 1️⃣MVTec数据集和预处理 MVTec数据集是一个工业缺陷检测的数据集,我这里有这么多类别 其中一类,img和label大概长这个样子 拿到数据集 首先定位问题属性,你拥有的资源(数据集,显卡,时间)有多少。你的需求是什么, 要多分类,还是不分类,要精度,要速度,还是要均衡。