apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Glenn Jocher
2019-07-20 01:28:29 +02:00
parent d1abe51876
commit d6edefa8ab
3 changed files with 34 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/utils.py
+3 -2
View File
@@ -534,11 +534,12 @@ def coco_only_people(path='../coco/labels/val2014/'):
print(labels.shape[0], file)
def select_best_evolve(path='../../Downloads/evolve*.txt'): # from utils.utils import *; select_best_evolve()
def select_best_evolve(path='evolve*.txt'): # from utils.utils import *; select_best_evolve()
# Find best evolved mutation
for file in sorted(glob.glob(path)):
x = np.loadtxt(file, dtype=np.float32)
print(file, x[x[:, 2].argmax()])
fitness = x[:, 2] * 0.5 + x[:, 3] * 0.5 # weighted mAP and F1 combination
print(file, x[fitness.argmax()])
def kmeans_targets(path='./data/coco_64img.txt'): # from utils.utils import *; kmeans_targets()