From 674681e1d6ab72a0550d53feeeaae34902d82af7 Mon Sep 17 00:00:00 2001 From: sinofis Date: Thu, 31 Dec 2020 17:28:37 +0100 Subject: [PATCH] reduce default training rotation range --- core/imagelib/warp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/imagelib/warp.py b/core/imagelib/warp.py index ac4f324..559c2e8 100644 --- a/core/imagelib/warp.py +++ b/core/imagelib/warp.py @@ -2,7 +2,7 @@ import numpy as np import cv2 from core import randomex -def gen_warp_params (w, flip, rotation_range=[-10,10], scale_range=[-0.5, 0.5], tx_range=[-0.05, 0.05], ty_range=[-0.05, 0.05], rnd_state=None ): +def gen_warp_params (w, flip, rotation_range=[-2,2], scale_range=[-0.5, 0.5], tx_range=[-0.05, 0.05], ty_range=[-0.05, 0.05], rnd_state=None ): if rnd_state is None: rnd_state = np.random