|
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!') |