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

GIoU to default

This commit is contained in:
glenn-jocher 2019-07-05 12:33:37 +02:00
parent 429bd3b8a9
commit 32a52dfb02
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -284,7 +284,7 @@ def print_mutation(hyp, results):
if opt.cloud_evolve: if opt.cloud_evolve:
os.system('gsutil cp gs://yolov4/evolve.txt .') # download evolve.txt os.system('gsutil cp gs://yolov4/evolve.txt .') # download evolve.txt
with open('evolve.txt', 'a') as f: # append result to evolve.txt with open('evolve.txt', 'a') as f: # append result
f.write(c + b + '\n') f.write(c + b + '\n')
os.system('gsutil cp evolve.txt gs://yolov4') # upload evolve.txt os.system('gsutil cp evolve.txt gs://yolov4') # upload evolve.txt
else: else:
@ -332,7 +332,7 @@ if __name__ == '__main__':
print_mutation(hyp, results) # Write mutation results print_mutation(hyp, results) # Write mutation results
for _ in range(gen): for _ in range(gen):
# Get best hyperparamters # Get best hyperparameters
x = np.loadtxt('evolve.txt', ndmin=2) x = np.loadtxt('evolve.txt', ndmin=2)
fitness = x[:, 2] * 0.9 + x[:, 3] * 0.1 # fitness as weighted combination of mAP and F1 fitness = x[:, 2] * 0.9 + x[:, 3] * 0.1 # fitness as weighted combination of mAP and F1
x = x[fitness.argmax()] # select best fitness hyps x = x[fitness.argmax()] # select best fitness hyps