|
|
@@ -5,8 +5,6 @@ sys.path.append("/ssd2/pengjuncai/PaddleRS")
|
|
|
import paddlers as pdrs
|
|
|
from paddlers import transforms as T
|
|
|
|
|
|
-# 定义训练和验证时的transforms
|
|
|
-# API说明：https://github.com/PaddlePaddle/PaddleX/blob/develop/docs/apis/transforms/transforms.md
|
|
|
train_transforms = T.Compose([
|
|
|
T.MixupImage(mixup_epoch=-1), T.RandomDistort(),
|
|
|
T.RandomExpand(im_padding_value=[123.675, 116.28, 103.53]), T.RandomCrop(),
|
|
|
@@ -22,8 +20,7 @@ eval_transforms = T.Compose([
|
|
|
mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
|
|
|
])
|
|
|
|
|
|
-# 定义训练和验证所用的数据集
|
|
|
-# API说明：https://github.com/PaddlePaddle/PaddleX/blob/develop/docs/apis/datasets.md
|
|
|
+
|
|
|
train_dataset = pdrs.datasets.VOCDetection(
|
|
|
data_dir='insect_det',
|
|
|
file_list='insect_det/train_list.txt',
|
|
|
@@ -38,15 +35,10 @@ eval_dataset = pdrs.datasets.VOCDetection(
|
|
|
transforms=eval_transforms,
|
|
|
shuffle=False)
|
|
|
|
|
|
-print(train_dataset[0])
|
|
|
|
|
|
-# 初始化模型，并进行训练
|
|
|
-# 可使用VisualDL查看训练指标，参考https://github.com/PaddlePaddle/PaddleX/blob/develop/docs/visualdl.md
|
|
|
num_classes = len(train_dataset.labels)
|
|
|
model = pdrs.tasks.det.PPYOLO(num_classes=num_classes, backbone='ResNet50_vd_dcn')
|
|
|
|
|
|
-# API说明：https://github.com/PaddlePaddle/PaddleX/blob/develop/docs/apis/models/detection.md
|
|
|
-# 各参数介绍与调整说明：https://github.com/PaddlePaddle/PaddleX/blob/develop/docs/parameters.md
|
|
|
model.train(
|
|
|
num_epochs=200,
|
|
|
train_dataset=train_dataset,
|
juncaipeng