Swin Transformer实战目标检测：训练自己的数据集

bai666ai 2023-01-13 00:38:19

课程名称适应人群
Swin Transformer实战目标检测：训练自己的数据集希望学习Swin Transformer目标检测的学员

Transformer发轫于NLP（自然语言处理），并跨界应用到CV（计算机视觉）领域。 Swin Transformer是基于Transformer的计算机视觉骨干网，在图像分类、目标检测、实例分割、语义分割等多项下游CV应用中取得了SOTA的性能。该项工作也获得了ICCV 2021顶会最佳论文奖。

 

本课程将手把手地教大家使用labelImg标注和使用Swin Transformer训练自己的数据集。 

 

本课程将介绍Transformer及在CV领域的应用、Swin Transformer的原理。 课程以多目标检测（足球和梅西同时检测）为例进行Swin Transformer实战演示。

 

课程在Windows和Ubuntu系统上分别做项目演示。包括：安装软件环境、安装Pytorch、安装Swin-Transformer-Object-Detection、标注自己的数据集、准备自己的数据集（自动划分训练集和验证集）、数据集格式转换（Python脚本完成）、修改配置文件、训练自己的数据集、测试训练出的网络模型、性能统计、日志分析。 

图片检测效果

 

相关课程： 

《Transformer原理与代码精讲（PyTorch）》https://edu.csdn.net/course/detail/36697

《Transformer原理与代码精讲（TensorFlow）》https://edu.csdn.net/course/detail/36699

《ViT（Vision Transformer）原理与代码精讲》https://edu.csdn.net/course/detail/36719

《DETR原理与代码精讲》https://edu.csdn.net/course/detail/36768

《Swin Transformer实战目标检测：训练自己的数据集》https://edu.csdn.net/course/detail/36585

《Swin Transformer实战实例分割：训练自己的数据集》https://edu.csdn.net/course/detail/36586

《Swin Transformer原理与代码精讲》 https://download.csdn.net/course/detail/37045

...全文
4529 52 打赏 收藏 转发到动态 举报
AI 作业
写回复
用AI写文章
52 条回复
切换为时间正序
请发表友善的回复…
发表回复
我要长高！ 2023-12-18
  • 打赏
  • 举报
 回复

D:\mmcv\mmcv\ops\csrc\pytorch_cuda_helper.hpp(4): fatal error C1083: 无法打开包括文件: “ATen/ATen.h”: No such file or directory
error: command 'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin\nvcc.exe' failed with exit code 2
我一直搞不懂，前面的指令都可以执行并且没有错误，但是执行到这条指令到时候，总是报这样的错误，非常无语

我要长高！ 2023-12-18
  • 打赏
  • 举报
 回复

D:\mmcv\mmcv\ops\csrc\pytorch_cuda_helper.hpp(4): fatal error C1083: 无法打开包括文件: “ATen/ATen.h”: No such file or directory
error: command 'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin\nvcc.exe' failed with exit code 2
python setup.py build_ext执行这句命令楚雄的错误，这是什么原因？

bai666ai 2023-07-13
  • 打赏
  • 举报
 回复

电脑上视频播放时，屏幕左上可看到“下载课件”的链接。先下载课件，项目流程的课件中有网盘链接，可下载其它课程资料。

阿偶～ 2023-03-10
  • 打赏
  • 举报
 回复

老师我训练到Epoch [10]最后的时候出现了一下问题·
OSError: [Errno 38] Function not implemented: 'epoch_10.pth' -> '/mnt/Swin-Transformer-Object-Detection/work_dirs/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco/latest.pth'

bai666ai 2023-07-13
  • 举报
 回复
@阿偶～ 根据错误信息，看起来这个问题出现在`mask_rcnn_swin_tiny_patch4_window7_1x.pth`模型的加载过程中，可能是由于模型文件与代码之间的不匹配导致的。 `TypeError: init() got an unexpected keyword argument 'embed_dim'`的错误提示表明模型的初始化过程中使用了一个意外的关键字参数 `embed_dim`。 这可能是由于以下原因之一导致的： 1. 模型文件不匹配：模型文件 `mask_rcnn_swin_tiny_patch4_window7_1x.pth` 可能与当前代码所使用的模型定义不兼容。可能是模型文件是用不同的代码版本或不同的模型配置文件进行训练得到的。 2. 模型定义不匹配：当前的代码文件 `image_demo.py` 可能使用了一个不同的模型定义或配置文件，而模型文件 `mask_rcnn_swin_tiny_patch4_window7_1x.pth` 中的模型与之不兼容。 为了解决这个问题，你可以尝试以下几个步骤： 1. 检查模型文件和代码版本：确保使用的模型文件是与代码版本兼容的，并且是通过相同的代码和配置文件训练得到的。如果模型文件是从其他来源获取的，可能需要确认其与当前代码的兼容性。 2. 确认模型定义和配置：检查代码中使用的模型定义和配置文件，确保其与模型文件中的模型定义和配置相匹配。确保代码中使用的模型与模型文件中的模型具有相同的结构和参数。 3. 更新代码或模型文件：如果你确定代码和模型文件之间存在版本不匹配或兼容性问题，可以尝试更新代码或使用适用于当前代码版本的模型文件。
Ysj78912 2023-02-15
  • 打赏
  • 举报
 回复

老师你好，麻烦问一下训练的batchsize是在哪个文件里设置的，我没有找到呢

bai666ai 2023-07-13
  • 举报
 回复
@Ysj78912 在训练命令中设置
Ysj78912 2023-02-11
  • 打赏
  • 举报
 回复 1

老师你好，我在安装mmcv第四步python setup.py build_ext 时遇到了问题，在anaconda ps prompt中报错RuntimeError: Error compiling objects for extension

Ysj78912 2023-02-11
  • 举报
 回复 1
@Ysj78912 已经解决了
一只蒜头青蛙 2023-05-16
  • 举报
 回复
@Ysj78912 是啥原因呀，宝子
u012420361 2022-05-31
  • 打赏
  • 举报
 回复
老师，https://github.com/SwinTransformer提供了百度网盘的权重下载地址，但我没找到密码
bai666ai 2023-07-13
  • 举报
 回复
@u012420361 电脑上视频播放时，屏幕左上可看到“下载课件”的链接。先下载课件，项目流程的课件中有网盘链接，可下载其它课程资料。
u012420361 2022-05-31
  • 打赏
  • 举报
 回复
老师，https://github.com/SwinTransformer提供了百度网盘的权重下载地址，但我没找到密码
bai666ai 2023-07-13
  • 举报
 回复
@u012420361 电脑上视频播放时，屏幕左上可看到“下载课件”的链接。先下载课件，项目流程的课件中有网盘链接，可下载其它课程资料。
u012420361 2022-05-31
  • 打赏
  • 举报
 回复
老师，https://github.com/SwinTransformer提供了百度网盘的权重下载地址，但我没找到密码
bai666ai 2023-07-13
  • 举报
 回复
@u012420361 电脑上视频播放时，屏幕左上可看到“下载课件”的链接。先下载课件，项目流程的课件中有网盘链接，可下载其它课程资料。
u012420361 2022-05-31
  • 打赏
  • 举报
 回复
老师，https://github.com/SwinTransformer提供了百度网盘的权重下载地址，但我没找到密码
u012420361 2022-05-31
  • 打赏
  • 举报
 回复
老师，https://github.com/SwinTransformer提供了百度网盘的权重下载地址，但我没找到密码
u012420361 2022-05-31
  • 打赏
  • 举报
 回复
老师，https://github.com/SwinTransformer提供了百度网盘的权重下载地址，但我没找到密码
补刀翁 2022-05-04
  • 打赏
  • 举报
 回复
老师你好，我在训练完成后进行测试性能时报出这样的错误，请问该如何解决？test.py: error: unrecognized arguments: work_dirs/faster_rcnn_swin_tiny_patch4_window7_mstrain_480- 800_adamw_1x_coco/latest.pth
bai666ai 2022-05-05
  • 举报
 回复
@补刀翁 <p>你的pytorch是什么版本？</p> <p> </p>
Asif Ahmed(姜明昊 ） 2023-01-26
  • 举报
 回复
@补刀翁 i had the same problem, try to set the path in write way(dont copy and paste from pdf document).Best of luck.
我要买辆兰博 2022-04-29
  • 打赏
  • 举报
 回复
老师您好，我点了好几次下载课件怎么都没反应呢？
bai666ai 2022-04-29
  • 举报
 回复
@我要买辆兰博 <p>换个浏览器试试。或与CSDN客服联系一下</p>
我要买辆兰博 2022-04-29
  • 打赏
  • 举报
 回复
点下载课件怎么没反应啊？
bai666ai 2022-04-29
  • 举报
 回复
@我要买辆兰博 <p>换个浏览器试试。或与CSDN客服联系一下</p>
补刀翁 2022-04-19
  • 打赏
  • 举报
 回复
老师，您好。我在Windows下搭建环境到安装mmcv第4步python setup.py build_ext时遇到问题 D:\ana\envs\swin1\mmcv\mmcv\ops\csrc\pytorch\cc_attention_cuda.cu(4): fatal error C1083: 无法打开包括文件: “THC/THC.h”: No such file or directory error: command 'C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.3\\bin\\nvcc.exe' failed with exit code 2不知道如何解决。
bai666ai 2022-04-19
  • 举报
 回复
@补刀翁 <p>你的GPU是什么型号？cuda是什么版本？</p>
haut-ljr 2022-04-18
  • 打赏
  • 举报
 回复
老师，您好，我在运行demo.py的时候老是遇到这个问题，image_demo.py: error: the following arguments are required: img, config, checkpoint。那个预训练权重文件也下载过放在目录下了，请问怎么解决啊。
bai666ai 2022-04-18
  • 举报
 回复
@haut-ljr <p>你按课程演示的视频看下是否缺少了命令参数，或者配置文件的问题</p> <p> </p>
wangyuhan555 2022-04-07
  • 打赏
  • 举报
 回复
老师你好，我用了课程提供的训练集，训练时依然是 tuple index out of range，请问怎么解决呢？
bai666ai 2022-04-07
  • 举报
 回复
@wangyuhan555 <p>检查下数据集放置目录是否正确</p> <p> </p>
weixin_49299669 2022-03-19
  • 打赏
  • 举报
 回复
老师，我在测试时报错：fused_weight_gradient_mlp_cuda` module not found. gradient accumulation fusion with weight gradient computation disabled. usage: image_demo.py [-h] [--device DEVICE] [--score-thr SCORE_THR] img config checkpoint image_demo.py: error: the following arguments are required: config, checkpoint
bai666ai 2022-03-19
  • 举报
 回复
@weixin_49299669 <p>检查下命令的参数是否正确</p>
wangyuhan555 2022-03-10
  • 打赏
  • 举报
 回复
老师您好，我在运行训练命令时出现了 tuple index out of range这个问题，而且work-dir里面的文件不全，请问怎么解决呢？
wangyuhan555 2022-03-10
  • 举报
 回复
@wangyuhan555 File "D:\wyhbysjrj\lib\site-packages\apex-0.1-py3.6.egg\apex\amp\utils.py", line 97, in cached_cast if cached_x.grad_fn.next_functions[1][0].variable is not x: IndexError: tuple index out of rang
bai666ai 2022-03-13
  • 举报
 回复
@wangyuhan555 <p>先用课程中的数据集试试有没有问题</p>
加载更多回复（7）
Swin Transformer实战：timm中的 Swin Transformer实现图像分类（多GPU）。
本例提取了植物幼苗数据集中的部分数据做数据集，数据集共有12种类别，演示如何使用timm版本的Swin Transformer图像分类模型实现分类任务已经对验证集得分的统计，本文实现了多个GPU并行训练。 通过本文你和学到： 1、如何从timm调用模型、loss和Mixup？ 2、如何制作ImageNet数据集？ 3、如何使用Cutout数据增强？ 4、如何使用Mixup数据增强。 5、如何实现多个GPU训练和验证。 6、如何使用余弦退火调整学习率？ 7、如何使用classification_report实现对模型的评价。 8、预测的两种写法。
Swin-T-使用C++实现Swin-Transformer目标检测算法-附项目源码-优质项目实战.zip
Swin-T_使用C++实现Swin-Transformer目标检测算法_附项目源码_优质项目实战
Swin-Transformer 图像分类网络实战项目：20常见小猫种图像分类数据集（迁移学习）
基于svit网络对20常见小猫种图像分类数据集的迁移学习项目，包含代码、数据集和训练好的权重文件，经过测试，可以直接运行 【网络】swin-transformer，参数量为8千万左右 【数据集】20常见小猫种图像分类数据集(布偶、波斯猫、蓝猫、橘猫等等） 【训练train.py】 1、训练过程中，会对数据集进行随机裁剪、翻转等数据增广等等。 2、网络初始化会自动载入官方在imagenet上的预训练权重进行迁移学习。 3、训练脚本会自动生成数据集类别的 json 文件，并且通过json文件自动设定网络的输出维度，不需要自己定义输出的channel。train.py 训练完成会生成训练集的loss曲线、学习率衰减曲线、测试集的accuracy曲线以及训练日志，保存在run_results文件内。 【预测predict.py】只需要将待预测的图像放在inference文件夹下，代码会自动将该文件下所有的图像进行预测，并在原图像左上角写入最大的前三个类别和概率，无需更改任何代码 【本项目仅仅训练了100个epoch，准确率为0.84，加大epoch可以提升网络精度】
YOLOV5 改进实战项目【更换backbone为Swin-Transformer】：农场乌鸡目标检测数据集
基于YOLOV5【更换backbone为Swin-Transformer】对农场乌鸡目标检测数据集的目标检测实战项目，包含代码、数据集、训练好的权重参数，经测试，代码可以直接使用 【数据集介绍】乌鸡图像数据，1类别：乌鸡 训练集datasets-images-train：400张图片和400个标签txt文件组成 测试集datasets-images-val：100张图片和100个标签txt文件组成 【yolov5】项目总大小：213MB 项目迭代了100个epoch，在runs目录下保存了训练结果，训练最好的精度map0.5=0.84，map0.5:0.95=0.42。训练过程中会生成验证集的混淆矩阵，PR曲线、F1曲线等等runs/detect目录下保存了网络推理训练集的全部结果，推理效果很好 更多yolov5改进介绍、或者如何训练，请参考： https://blog.csdn.net/qq_44886601/category_12605353.html
Swin-Transformer 图像分类网络实战项目：MIT-BIH心电图像分类数据集迁移学习项目
基于Swin-Transformer网络对MIT-BIH心电图像分类数据集迁移学习项目，包含代码、数据集和训练好的权重文件，可直接运行。 这里不是数据的全部，原数据转换后的图像有9W张，这里只使用了一小部分子集来训练 项目总大小：971MB 本数据集分为以下5类：A、L、N、R、V 下载解压后的图像目录：训练集（3186张图片）、和测试集（793张图片） data-train 训练集-每个子文件夹放同类别的图像，文件夹名为分类类别 data-test 测试集-每个子文件夹放同类别的图像，文件夹名为分类类别 网络训练的时候采用cos 学习率自动衰减，训练了20个epoch。模型在测试集最好的表现达到90%精度！！！在run_results 目录下存有最好的权重文件，以及训练日志和loss、精度曲线等等 预测的时候，只需要运行predict即可，代码会自动将inference下所有图片推理，并取前三个概率最大类别的绘制在左上角 如果想要训练自己的数据集，请查看README文件
白老师的课程社区_NO_6

2

社区成员

173

社区内容

发帖
与我相关
我的任务
社区描述
人工智能技术专家
社区管理员
  • bai666ai
加入社区
  • 近7日
  • 近30日
  • 至今

加载中

查看更多榜单
社区公告
暂无公告

试试用AI创作助手写篇文章吧

+ 用AI写文章