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

updates

This commit is contained in:
Glenn Jocher 2020-01-17 18:05:28 -08:00
parent a4b8815ed9
commit bab855507a
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -221,7 +221,7 @@ def train():
# Prebias # Prebias
if prebias: if prebias:
if epoch < 20: # prebias if epoch < 1: # prebias
ps = 0.1, 0.9 # prebias settings (lr=0.1, momentum=0.9) ps = 0.1, 0.9 # prebias settings (lr=0.1, momentum=0.9)
else: # normal training else: # normal training
ps = hyp['lr0'], hyp['momentum'] # normal training settings ps = hyp['lr0'], hyp['momentum'] # normal training settings

8
utils/datasets.py
View File

@ -593,10 +593,10 @@ def load_mosaic(self, index):
# Augment # Augment
# img4 = img4[s // 2: int(s * 1.5), s // 2:int(s * 1.5)] # center crop (WARNING, requires box pruning) # img4 = img4[s // 2: int(s * 1.5), s // 2:int(s * 1.5)] # center crop (WARNING, requires box pruning)
img4, labels4 = random_affine(img4, labels4, img4, labels4 = random_affine(img4, labels4,
degrees=self.hyp['degrees'] * 0, degrees=self.hyp['degrees'] * 1,
translate=self.hyp['translate'] * 0, translate=self.hyp['translate'] * 1,
scale=self.hyp['scale'] * 0, scale=self.hyp['scale'] * 1,
shear=self.hyp['shear'] * 0, shear=self.hyp['shear'] * 1,
border=-s // 2) # border to remove border=-s // 2) # border to remove
return img4, labels4 return img4, labels4