From 5dc64138e439247b4e707dd6cebfe19d8d77f5b1 Mon Sep 17 00:00:00 2001 From: lyuwenyu Date: Tue, 23 Jan 2024 11:17:54 +0800 Subject: [PATCH] align data aug --- rtdetr_pytorch/configs/rtdetr/include/dataloader.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rtdetr_pytorch/configs/rtdetr/include/dataloader.yml b/rtdetr_pytorch/configs/rtdetr/include/dataloader.yml index a4d5f049..e3e6bc1f 100644 --- a/rtdetr_pytorch/configs/rtdetr/include/dataloader.yml +++ b/rtdetr_pytorch/configs/rtdetr/include/dataloader.yml @@ -6,7 +6,7 @@ train_dataloader: return_masks: False transforms: ops: - - {type: RandomPhotometricDistort, p: 0.8} + - {type: RandomPhotometricDistort, p: 0.5} - {type: RandomZoomOut, fill: 0} - {type: RandomIoUCrop, p: 0.8} - {type: SanitizeBoundingBox, min_size: 1}