Skip to content

Commit

Permalink
Add new model
Browse files Browse the repository at this point in the history
  • Loading branch information
mathieuboudreau committed Nov 11, 2021
1 parent 316a495 commit d7fb180
Show file tree
Hide file tree
Showing 14 changed files with 181 additions and 116 deletions.
2 changes: 0 additions & 2 deletions default_SEM_model/checkpoint

This file was deleted.

108 changes: 0 additions & 108 deletions default_SEM_model/config_network.json

This file was deleted.

5 changes: 0 additions & 5 deletions default_SEM_model/data_test/README.txt

This file was deleted.

Binary file removed default_SEM_model/data_test/image.png
Binary file not shown.
Binary file removed default_SEM_model/data_test/mask.png
Binary file not shown.
1 change: 0 additions & 1 deletion default_SEM_model/data_test/pixel_size_in_micrometer.txt

This file was deleted.

Binary file not shown.
Binary file removed default_SEM_model/model.ckpt.data-00000-of-00001
Binary file not shown.
Binary file removed default_SEM_model/model.ckpt.index
Binary file not shown.
Binary file removed default_SEM_model/model.ckpt.meta
Binary file not shown.
Binary file removed default_SEM_model/model.hdf5
Binary file not shown.
181 changes: 181 additions & 0 deletions model_seg_rat_axon-myelin_sem/model_seg_rat_axon-myelin_sem.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,181 @@
{
"command": "train",
"gpu_ids": [
7
],
"path_output": "../data_extrassd_maboudb/20210816_optim_sem/20211010_log_sem-loss-name-DiceLoss-scheduler-initial_lr-0.005-lr_scheduler-name-CosineAnnealingLR",
"model_name": "model_seg_rat_axon-myelin_sem",
"debugging": true,
"log_file": "log",
"object_detection_params": {
"object_detection_path": null,
"safety_factor": [
1.0,
1.0,
1.0
],
"gpu_ids": 7,
"path_output": "../data_extrassd_maboudb/20210816_optim_sem/20211010_log_sem-loss-name-DiceLoss-scheduler-initial_lr-0.005-lr_scheduler-name-CosineAnnealingLR"
},
"loader_parameters": {
"path_data": [
"../data_extrassd_maboudb/20210816_optim_sem/data_axondeepseg_sem/"
],
"subject_selection": {
"n": [],
"metadata": [],
"value": []
},
"target_suffix": [
"_seg-axon-manual",
"_seg-myelin-manual"
],
"extensions": [
".png"
],
"roi_params": {
"suffix": null,
"slice_filter_roi": null
},
"contrast_params": {
"training_validation": [
"SEM"
],
"testing": [
"SEM"
],
"balance": {}
},
"slice_filter_params": {
"filter_empty_mask": false,
"filter_empty_input": true
},
"slice_axis": "axial",
"multichannel": false,
"soft_gt": false,
"is_input_dropout": false,
"bids_config": "ivadomed/config/config_bids.json"
},
"split_dataset": {
"fname_split": "../data_extrassd_maboudb/20210816_optim_sem/config/split_datasets.joblib",
"random_seed": 1461,
"split_method": "sample_id",
"data_testing": {
"data_type": "sample_id",
"data_value": [
"sample-data15"
]
},
"balance": null,
"train_fraction": 0.7,
"test_fraction": 0.1
},
"training_parameters": {
"batch_size": 4,
"loss": {
"name": "DiceLoss"
},
"training_time": {
"num_epochs": 150,
"early_stopping_patience": 50,
"early_stopping_epsilon": 0.001
},
"scheduler": {
"initial_lr": 0.005,
"lr_scheduler": {
"name": "CosineAnnealingLR"
}
},
"balance_samples": {
"applied": false,
"type": "gt"
},
"mixup_alpha": null,
"transfer_learning": {
"retrain_model": null,
"retrain_fraction": 1.0,
"reset": true
}
},
"default_model": {
"name": "Unet",
"dropout_rate": 0.2,
"bn_momentum": 0.1,
"depth": 4,
"is_2d": true,
"final_activation": "sigmoid",
"length_2D": [
256,
256
],
"stride_2D": [
244,
244
]
},
"uncertainty": {
"epistemic": false,
"aleatoric": false,
"n_it": 0
},
"postprocessing": {
"binarize_maxpooling": {}
},
"evaluation_parameters": {},
"transformation": {
"Resample": {
"wspace": 0.0001,
"hspace": 0.0001
},
"RandomAffine": {
"degrees": 2.5,
"scale": [
0.05,
0.05
],
"translate": [
0.015,
0.015
],
"applied_to": [
"im",
"gt"
],
"dataset_type": [
"training"
]
},
"ElasticTransform": {
"alpha_range": [
100.0,
150.0
],
"sigma_range": [
4.0,
5.0
],
"p": 0.5,
"applied_to": [
"im",
"gt"
],
"dataset_type": [
"training"
]
},
"NormalizeInstance": {
"applied_to": [
"im"
]
}
},
"training_sha256": {
"sub-rat3_sample-data10_SEM.png": "d402873851fa675b0bbaeff8ab752e92e0b8159834f416c4428f78a15883f5ce",
"sub-rat3_sample-data9_SEM.png": "067af38d7f98cb7eef988d38c8821af7dc33c73146e79eeff66dc9349a336c47",
"sub-rat4_sample-data12_SEM.png": "d121f05fe9f03458b4dd9a9e4f0fdeba64e5cbd97ec4584415931144e012b764",
"sub-rat2_sample-data5_SEM.png": "e878e6d412e0a674a67f57f983a97eac70edc41e04ab222ef4338d2653377a95",
"sub-rat8_sample-V915_SEM.png": "433687677047cc3a2966a27c35325eaef89420fa0b0301d017892f70e3d87fdd",
"sub-rat7_sample-Maxo03img60_SEM.png": "952014851a21ad3cce229d8ba72ea935ad762bbe3bfb48e5f5b972971f57f566",
"sub-rat3_sample-data11_SEM.png": "a3fac42bb940fe20371167bc64bae94e47f8551dd699357122c161b39d9fa9e1"
}
}
Binary file not shown.
Binary file not shown.

0 comments on commit d7fb180

Please sign in to comment.