Initial commit
This commit is contained in:
33
Tensorflow/tutoriel27/config.py
Normal file
33
Tensorflow/tutoriel27/config.py
Normal file
@@ -0,0 +1,33 @@
|
||||
import numpy as np
|
||||
|
||||
dict={'leukocyte': (255, 255, 0 ),
|
||||
'red blood cell':(0 , 0 , 255),
|
||||
'ring': (0 , 255, 0 ),
|
||||
'schizont': (255, 0 , 255),
|
||||
'trophozoite': (255, 0 , 0 ),
|
||||
'difficult': (0 , 0 , 0 ),
|
||||
'gametocyte': (0 , 255, 255)}
|
||||
dict2=[]
|
||||
for d in dict:
|
||||
dict2.append(d)
|
||||
|
||||
largeur=256
|
||||
hauteur=192
|
||||
cellule_x=16
|
||||
cellule_y=12
|
||||
nbr_classes=len(dict)
|
||||
r_x=int(largeur/cellule_x)
|
||||
r_y=int(hauteur/cellule_y)
|
||||
max_objet=60
|
||||
|
||||
anchors=np.array([[3.0, 1.5], [2.0, 2.0], [1.5, 3.0]])
|
||||
nbr_boxes=len(anchors)
|
||||
|
||||
batch_size=16
|
||||
|
||||
lambda_coord=5
|
||||
lambda_noobj=0.5
|
||||
#lambda_coord=1
|
||||
#lambda_noobj=1
|
||||
|
||||
seuil_iou_loss=0.6
|
||||
Reference in New Issue
Block a user