Segmentation of COVID-19 Chest CT Scans using TransUNet
Transformers have shown great performance in the field of Image Segmentation. It would be amazing to train them using JAX/Flax. The goal of this project is automatic COVID-19 lesion segmentation from chest CT.

2. Language
The model will be trained in English.

3. Model
The TransUNet model will be used for segmentation.

4. Datasets
Possible links to publicly available datasets include:


dataset-01: https://doi.org/10.5281/zenodo.3757476 4


dataset-02: COVID-19 - Medical segmentation 4



5. Training scripts
Are there publicly available training scripts that can be used/tweaked for the project?
Answer: Unfortunately, No.

6. (Optional) Challenges

Build TransUnet model in JAX/Flax
Implement Augmentation for Segmentation.


7. (Optional) Desired project outcome

This project will help clinicians segment COVID-19 lesions which will help clinicians identify potential COVID-19 cases.

TransUNet can be later on other segmentation tasks.