36 lines
1.3 KiB
Plaintext
36 lines
1.3 KiB
Plaintext
|
STDC语义分割模型
|
|||
|
|
1.通过data里面的json文件，修改任务, 配置文件如下：
|
|||
|
|
"dspth":"../../data/RoadLane/", #数据文件夹
|
|||
|
|
"cropsize":"1280,720", #模型的宽、高
|
|||
|
|
"labelJson":"./data/RoadLane_info.json",#标签的信息
|
|||
|
|
"n_classes":3,#语义分割的类别叔叔
|
|||
|
|
"ignore_idx":255 #忽略的类别数
|
|||
|
|
2. 数据组织
|
|||
|
|
├── train
|
|||
|
|
│ ├── images []
|
|||
|
|
│ ├── labels []
|
|||
|
|
│ └── t.txt
|
|||
|
|
└── val
|
|||
|
|
├── images []
|
|||
|
|
└── labels []
|
|||
|
|
图像放在images下面，标签放在labels下面。
|
|||
|
|
3. 标签格式 及数据说明文件
|
|||
|
|
labels里的数据都是png格式，里面放的是RGB彩色标签。如：道路-（256,0,0）表示
|
|||
|
|
数据说明文件：./data/RoadLane_info.json， 一般放在./data 文件下面
|
|||
|
|
{
|
|||
|
|
"hasInstances": false,
|
|||
|
|
"category": "void",
|
|||
|
|
"catid": 0,
|
|||
|
|
"name": "speedRoad",
|
|||
|
|
"ignoreInEval": true,
|
|||
|
|
"id":1,
|
|||
|
|
"color": [
|
|||
|
|
128,
|
|||
|
|
0,
|
|||
|
|
0
|
|||
|
|
],
|
|||
|
|
主要是"id"和"color"要对应上，"id"是从0开始编号
|
|||
|
|
4. 模型训练
|
|||
|
|
python train.py --parJson ./data/RoadLane.json --respath ./checkpooints/0430pm --gpuId 0
|
|||
|
|
# ./checkpooints/0430pm --为之前保存的训练路径
|