|
@@ -8,7 +8,7 @@ pdx.utils.download_and_decompress(dataset, path='./')
|
|
|
# 定义训练和验证时的transforms
|
|
# 定义训练和验证时的transforms
|
|
|
# API说明:https://github.com/PaddlePaddle/PaddleX/blob/release/2.0-rc/paddlex/cv/transforms/operators.py
|
|
# API说明:https://github.com/PaddlePaddle/PaddleX/blob/release/2.0-rc/paddlex/cv/transforms/operators.py
|
|
|
train_transforms = T.Compose([
|
|
train_transforms = T.Compose([
|
|
|
- T.MixupImage(mixup_epoch=250), T.RandomDistort(),
|
|
|
|
|
|
|
+ T.MixupImage(mixup_epoch=-1), T.RandomDistort(),
|
|
|
T.RandomExpand(im_padding_value=[123.675, 116.28, 103.53]), T.RandomCrop(),
|
|
T.RandomExpand(im_padding_value=[123.675, 116.28, 103.53]), T.RandomCrop(),
|
|
|
T.RandomHorizontalFlip(), T.BatchRandomResize(
|
|
T.RandomHorizontalFlip(), T.BatchRandomResize(
|
|
|
target_sizes=[320, 352, 384, 416, 448, 480, 512, 544, 576, 608],
|
|
target_sizes=[320, 352, 384, 416, 448, 480, 512, 544, 576, 608],
|
|
@@ -46,14 +46,15 @@ model = pdx.det.PPYOLO(num_classes=num_classes, backbone='ResNet50_vd_dcn')
|
|
|
# API说明:https://github.com/PaddlePaddle/PaddleX/blob/release/2.0-rc/paddlex/cv/models/detector.py#L155
|
|
# API说明:https://github.com/PaddlePaddle/PaddleX/blob/release/2.0-rc/paddlex/cv/models/detector.py#L155
|
|
|
# 各参数介绍与调整说明:https://paddlex.readthedocs.io/zh_CN/develop/appendix/parameters.html
|
|
# 各参数介绍与调整说明:https://paddlex.readthedocs.io/zh_CN/develop/appendix/parameters.html
|
|
|
model.train(
|
|
model.train(
|
|
|
- num_epochs=405,
|
|
|
|
|
|
|
+ num_epochs=200,
|
|
|
train_dataset=train_dataset,
|
|
train_dataset=train_dataset,
|
|
|
train_batch_size=8,
|
|
train_batch_size=8,
|
|
|
eval_dataset=eval_dataset,
|
|
eval_dataset=eval_dataset,
|
|
|
|
|
+ pretrain_weights='COCO',
|
|
|
learning_rate=0.005 / 12,
|
|
learning_rate=0.005 / 12,
|
|
|
- warmup_steps=1000,
|
|
|
|
|
|
|
+ warmup_steps=500,
|
|
|
warmup_start_lr=0.0,
|
|
warmup_start_lr=0.0,
|
|
|
save_interval_epochs=5,
|
|
save_interval_epochs=5,
|
|
|
- lr_decay_epochs=[243, 324],
|
|
|
|
|
|
|
+ lr_decay_epochs=[85, 135],
|
|
|
save_dir='output/ppyolo_r50vd_dcn',
|
|
save_dir='output/ppyolo_r50vd_dcn',
|
|
|
use_vdl=True)
|
|
use_vdl=True)
|