Diff of /bc-count/config.py [000000] .. [0be6a8]

Switch to unified view

a b/bc-count/config.py
1
##############################################
2
#                                            #
3
#         Project configuration file         #
4
#                                            #
5
# Author: Amine Neggazi                      #
6
# Email: neggazimedlamine@gmail/com          #
7
# Nick: nemo256                              #
8
#                                            #
9
# Please read bc-count/LICENSE               #
10
#                                            #
11
##############################################
12
13
''' 
14
Cell type can be wither rbc, wbc or plt
15
which stands for:
16
    rbc --> Red blood cells
17
    wbc --> White blood cells
18
    plt --> Platelets
19
'''
20
cell_type  = 'wbc'             # rbc, wbc or plt
21
model_type = 'do_unet'          # do_unet or segnet
22
23
if model_type == 'do_unet':
24
    model_name   = cell_type
25
    input_shape  = (188, 188, 3)
26
    output_shape = (100, 100, 1)
27
else:
28
    model_name   = cell_type + '_segnet'
29
    input_shape  = (128, 128, 3)
30
    output_shape = (128, 128, 1)
31
32
padding = [200, 100]
33
output_directory = 'output/' + model_type + '/' + cell_type
34
35
if not cell_type in ['rbc', 'wbc', 'plt']:
36
    print('Invalid cell type!')