3D Brain Tumor Segmentation

• Tech Stack: Python, PyTorch, Vision Transformer, 3D Computer Vision, Point Cloud Visualization
• Github URL: Project Link

Leveraged PyTorch to train Vnet and Unetr models, developing a 3D brain tumor segmentation algorithm on BRaTS2021 dataset

Optimized GPU usage in training and inference pipelines by 40% using PyTorch profiler and TensorBoard.

Utilized Matplotlib for interactive point cloud visualizations, aiding thorough analysis of predictions in 3D.