mmsegmention工程配置

mmsegmention工程配置:

  1. 下载模型权重
  2. 准备数据、并转换数据为txt
  3. Config/_base_/detasets/pascal_voc12.py
  1. 修改data_root 数据路径
  2. Samples_per_gpu=  和 workers_per_gpu= 如单gpu两个值一样2或4。
  1. mmseg/datasets/voc.py

1)修改CLASSES=种类和PALETTE=色盘

5、tool/train.py

1)修改‘config’对应模型;‘work-dir’保存日志权重路径在当前目录;‘load-from预训练权重,和config一一对应

6、config/_base_/models/对应py文件

1)Norm_cfg =dict(type=‘SyncBN’,requires_grad=True)单GPU改SyncBN为BN

2)所有的(两个)num_classes=改为自己的种类(不包含背景)

7、config/

1)对应模型脚本里面的‘../_base_/datasets/自己对应.py

2)这个脚本有的同样和config/_base_/models/对应py文件,会出现SyncBN和num_classes=,要对应修改。

8、预训练模型选择要找对应分辨率的模型

9、运行train.py

多卡训练改只动dish_train.sh里面的GPU=n(n为GPU数量)

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

慕容洛凝

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值