Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-10-09 02:10:25 +02:00
parent a4e9aa34ef
commit 500a798787
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
utils/datasets.py
View File

@ -444,6 +444,9 @@ class LoadImagesAndLabels(Dataset): # for training/testing
labels[:, 4] = ratio[1] * h * (x[:, 2] + x[:, 4] / 2) + padh labels[:, 4] = ratio[1] * h * (x[:, 2] + x[:, 4] / 2) + padh
if self.augment: if self.augment:
# Augment colorspace
augment_hsv(img, hgain=self.hyp['hsv_h'], sgain=self.hyp['hsv_s'], vgain=self.hyp['hsv_v'])
# Augment imagespace # Augment imagespace
g = 0.0 if mosaic else 1.0 # do not augment mosaics g = 0.0 if mosaic else 1.0 # do not augment mosaics
hyp = self.hyp hyp = self.hyp
@ -511,11 +514,6 @@ def load_image(self, index):
if self.augment and r < 1: # if training (NOT testing), downsize to inference shape if self.augment and r < 1: # if training (NOT testing), downsize to inference shape
h, w, _ = img.shape h, w, _ = img.shape
img = cv2.resize(img, (int(w * r), int(h * r)), interpolation=cv2.INTER_LINEAR) # _LINEAR fastest img = cv2.resize(img, (int(w * r), int(h * r)), interpolation=cv2.INTER_LINEAR) # _LINEAR fastest
# Augment colorspace
if self.augment:
augment_hsv(img, hgain=self.hyp['hsv_h'], sgain=self.hyp['hsv_s'], vgain=self.hyp['hsv_v'])
return img return img