Repository files navigation cell-cycle-classification
Modify saving directory like checkpoint, record and model via config.py
Step 1: Mitochondrai and nucleus prediction from brightfield
Set1 Input: brightfield | Label: mitochondria
Set2 Input: brightfield | Label: nucleus
Model: UNet
Optimizer: Adam
Loss function: Huber Loss
Accuracy: Pearson correlation coefficient
Batch size: 2
Epoch: 100
cd train
python train_predict_mito_nuclei.py
UNet Training result
Mitochondria
Training Pearson Correlation Coefficient: 0.78
Validation Pearson Correlation Coefficient: 0.72
nucleous
Training Pearson Correlation Coefficient: 0.85
Validation Pearson Correlation Coefficient: 0.83
cd plot
python predict_for_mitochondira_nucleus.py
python ploy_record.py
Step 2: Cell cycle classification from brightfield/predicted mitochondria/predicted nucleus
Input: brightfield + predicited mitochondria + predicted nucleus
Label: 4 class for each pixel (0: background, 1: RFP, 2: GFP, 3: Both)
Model: Efficient UNet
Optimizer: Adam
Loss function: Focal + Dice loss
Accuracy: F1 score
Batch size: 4
Epoch: 100
cd train
python train_cellcycle_phase.py
You can’t perform that action at this time.