模型训练
文章平均质量分 51
athrunsunny
这个作者很懒,什么都没留下…
展开
-
SG Former实战:训练自定义分类数据集
源码中数据集为Imagenet-1K,数据量比较大,完整的数据集容量超过100G,光是下载这个数据集就得花不少时间,为了玩转SG Former可以换成自定义的较小的数据集看看效果。直接将源码中labeled_memcached_dataset.py的McDataset()基础上进行修改。首先按照大多分类数据集的制作方式,文件夹名作为类别名,文件夹的结构和imagenet-1k的的一致。同时在main.py中--data设置数据集的路径./datasets。原创 2023-10-07 10:38:33 · 389 阅读 · 0 评论 -
RuntimeError: Address already in use pytorch分布式训练
出现该错误原因是通信端口被占用。之后重新训练即可正常。原创 2023-02-23 09:30:51 · 1637 阅读 · 0 评论 -
DAMOYOLO windows 单卡训练
最近达摩院放出了目前最能打的yolo算法,时间和精度都得到了提升 目前代码已经开源: 代码预设仅支持分布式训练,对于硬件资源有限的小伙伴来说,算法的训练就不是太友好了,但是对于想要尝试的小伙伴还是有办法的1、增加 否则会报:2、windows不支持nccl backbone所以init_process_group中改为‘gloo’二、改配置configs\xxx.py找到代码17行的ps:建议设置为8, 训练过程中占用显存较大三、改数据集路径damo\config\paths_catal原创 2022-12-04 12:53:15 · 1440 阅读 · 9 评论 -
yolov5 multi-task 检测人脸landmark
文件夹中创建三个文件(predict,test,train_landmark),类似yolov5的分类和语义分割任务。最后在utils中增加facedataloaders.py用于数据集的读取。克隆下最新的yolov5,在根目录下创建landmark文件夹。utils/plots.py还有个函数需要修改。添加完以上三个文件后需要对源码进行一些修改。同时在以下两个函数中增加LmkDetect。1、DetectionModel中。五个关键点(眼睛,鼻子,嘴)2、parse_model中。原创 2022-10-19 10:57:51 · 1732 阅读 · 2 评论 -
yolov6训练yolov5格式数据集
YOLOv6\yolov6\data\datasets.py中268行。YOLOv6\yolov6\core\engine.py中48行。主要还是懒得修改数据集,大部分数据集都是用的yolov5格式的。以上两处修改之后yolov6就可以直接使用yolov5数据集。在使用源码读取的时候会读不到labels的文件夹。修改如下py文件的代码。原创 2022-09-22 20:13:39 · 1082 阅读 · 0 评论 -
yolov7训练自定义数据集时的注意事项
yolov7的数据集格式和yolov5是一样的,基本上直接将yolov5的数据集拿过来用即可。还有点需要注意就是用yolov5训练后的cache文件,在训练yolov7是要删除。区别就是没了path,主要是有些数据比较大,不想移来移去,所以直接修改v7的代码。主要修改的是yolov7\utils\general.py。这样就只需要在配置文件中改数据集路径即可。区别就在yaml文件上,......原创 2022-08-12 17:06:39 · 4111 阅读 · 3 评论 -
windows swin transformer训练自己的目标检测数据集
主要是有几个地方的文件要修改一下config/swin下的配置文件,我用的是mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco.py_base_ = [ '../_base_/models/mask_rcnn_swin_fpn.py', # '../_base_/datasets/coco_instance.py', '../_base_/datasets/coco_detection.py',原创 2022-04-28 11:52:07 · 1194 阅读 · 0 评论 -
windows cpu训练swin transformer
主要是有几个地方的文件要修改一下config/swin下的配置文件,我用的是mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco.py_base_ = [ '../_base_/models/mask_rcnn_swin_fpn.py', '../_base_/datasets/coco_instance.py', # '../_base_/datasets/coco_detection.py',原创 2022-04-28 11:36:59 · 1391 阅读 · 0 评论 -
本地通过tensorboard监控服务器的训练过程
在服务器上直接使用tensorboard --logdir ./path 显示的ip是无法直接在本地端的网页打开的,需要设置ip和端口号,端口号最好是没被占用的可以使用如下指令:tensorboard --logdir ./path --port port --host ip如tensorboard --logdir ./log --port 8014 --host 127.0.0.1...原创 2022-04-28 09:07:52 · 664 阅读 · 0 评论 -
yolov5关闭wandb
yolov5训练过程中wandb总是提示登入账号,不登入还不能继续训练在\yolov5\utils\loggers\wandb\wandb_utils.py中import wandb的下面加上wandb = Nonetry: import wandb from wandb import init, finishexcept ImportError: wandb = None# 添加以下语句wandb = None同时在\yolov5\utils\loggers原创 2022-03-11 16:44:12 · 5644 阅读 · 1 评论