updates
This commit is contained in:
parent
e81c1ab501
commit
59cf3978fc
|
@ -230,7 +230,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
|
||||||
augment_hsv = True
|
augment_hsv = True
|
||||||
if self.augment and augment_hsv:
|
if self.augment and augment_hsv:
|
||||||
# SV augmentation by 50%
|
# SV augmentation by 50%
|
||||||
fraction = 0.50 # must be < 1.0
|
fraction = 0.25 # must be < 1.0
|
||||||
img_hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV) # hue, sat, val
|
img_hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV) # hue, sat, val
|
||||||
S = img_hsv[:, :, 1].astype(np.float32) # saturation
|
S = img_hsv[:, :, 1].astype(np.float32) # saturation
|
||||||
V = img_hsv[:, :, 2].astype(np.float32) # value
|
V = img_hsv[:, :, 2].astype(np.float32) # value
|
||||||
|
|
Loading…
Reference in New Issue