Unverified 提交 bf209f6f authored 作者: Mai Thanh Minh's avatar Mai Thanh Minh 提交者: GitHub

Skip HSV augmentation when hyperparameters are [0, 0, 0] (#3686)

* Create shortcircuit in augment_hsv when hyperparameter are zero * implement faster opt-in Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 814806c6
......@@ -632,6 +632,7 @@ def load_image(self, index):
def augment_hsv(img, hgain=0.5, sgain=0.5, vgain=0.5):
if hgain or sgain or vgain:
r = np.random.uniform(-1, 1, 3) * [hgain, sgain, vgain] + 1 # random gains
hue, sat, val = cv2.split(cv2.cvtColor(img, cv2.COLOR_BGR2HSV))
dtype = img.dtype # uint8
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论