|
@@ -22,10 +22,13 @@ MyDataset/ # 语义分割数据集根目录
|
|
|
## 划分训练集验证集
|
|
## 划分训练集验证集
|
|
|
|
|
|
|
|
**为了用于训练,我们需要在`MyDataset`目录下准备`train_list.txt`, `val_list.txt`和`labels.txt`三个文件**,分别用于表示训练集列表,验证集列表和类别标签列表。[点击下载语义分割示例数据集](https://bj.bcebos.com/paddlex/datasets/optic_disc_seg.tar.gz)
|
|
**为了用于训练,我们需要在`MyDataset`目录下准备`train_list.txt`, `val_list.txt`和`labels.txt`三个文件**,分别用于表示训练集列表,验证集列表和类别标签列表。[点击下载语义分割示例数据集](https://bj.bcebos.com/paddlex/datasets/optic_disc_seg.tar.gz)
|
|
|
|
|
+
|
|
|
|
|
+<!--
|
|
|
> 注:也可使用PaddleX自带工具,对数据集进行随机划分,**在数据集按照上面格式组织后**,使用如下命令即可快速完成数据集随机划分,其中split指标训练集的比例,剩余的比例用于验证集。
|
|
> 注:也可使用PaddleX自带工具,对数据集进行随机划分,**在数据集按照上面格式组织后**,使用如下命令即可快速完成数据集随机划分,其中split指标训练集的比例,剩余的比例用于验证集。
|
|
|
> ```
|
|
> ```
|
|
|
> paddlex --split_dataset --from Seg --pics ./JPEGImages --annotations ./Annotations --split 0.8 --save_dir ./splited_dataset_dir
|
|
> paddlex --split_dataset --from Seg --pics ./JPEGImages --annotations ./Annotations --split 0.8 --save_dir ./splited_dataset_dir
|
|
|
> ```
|
|
> ```
|
|
|
|
|
+-->
|
|
|
|
|
|
|
|
**labels.txt**
|
|
**labels.txt**
|
|
|
|
|
|
|
@@ -58,8 +61,18 @@ val_list列出用于验证时的图片集成,与其对应的标注文件,格
|
|
|
import paddlex as pdx
|
|
import paddlex as pdx
|
|
|
from paddlex.seg import transforms
|
|
from paddlex.seg import transforms
|
|
|
|
|
|
|
|
-train_transforms = transforms.ComposedSegTransforms(mode='train', train_crop_size=[512, 512])
|
|
|
|
|
-eval_transforms = transforms.ComposedSegTransforms(mode='eval', train_crop_size=[512, 512])
|
|
|
|
|
|
|
+train_transforms = transforms.Compose([
|
|
|
|
|
+ transforms.RandomHorizontalFlip(),
|
|
|
|
|
+ transforms.ResizeRangeScaling(),
|
|
|
|
|
+ transforms.RandomPaddingCrop(crop_size=512),
|
|
|
|
|
+ transforms.Normalize()
|
|
|
|
|
+])
|
|
|
|
|
+
|
|
|
|
|
+eval_transforms = transforms.Compose([
|
|
|
|
|
+ transforms.ResizeByLong(long_size=512),
|
|
|
|
|
+ transforms.Padding(target_size=512),
|
|
|
|
|
+ transforms.Normalize()
|
|
|
|
|
+])
|
|
|
|
|
|
|
|
train_dataset = pdx.datasets.SegDataset(
|
|
train_dataset = pdx.datasets.SegDataset(
|
|
|
data_dir='./MyDataset',
|
|
data_dir='./MyDataset',
|
|
@@ -71,5 +84,4 @@ eval_dataset = pdx.datasets.SegDataset(
|
|
|
file_list='./MyDataset/val_list.txt',
|
|
file_list='./MyDataset/val_list.txt',
|
|
|
label_list='MyDataset/labels.txt',
|
|
label_list='MyDataset/labels.txt',
|
|
|
transforms=eval_transforms)
|
|
transforms=eval_transforms)
|
|
|
-
|
|
|
|
|
```
|
|
```
|