Brain Glioma Classification

A simple classification model built with PyTorch, Torch Vision, and OpenCV that is able to detect up to 4 different types of Brain Gliomas to a 93.4% accuracy!

Like 3

Comment

A sample training image.

Inspiration

Inspired by AI prognosis in healthcare!

What it does

Trains model on training dataset and validates it on testing set.

How I built it

Documentation and Stack Overflow.

Challenges I ran into

Virtual Environments are Hard :(

Accomplishments that I'm proud of

90% accuracy for the model.

What I learned

Start project earlier next time.

What's next for Brain Glioma Classification

Increasing accuracy with epochs!