StenosisSeg

Tim Jing, Andy Wang, and Emily Liu's CS 231N Project

Table of Contents

• Dependencies
• Installation
• Training

Dependencies

The project requires the following dependencies:

• Python 3.9
• numpy
• matplotlib
• timm
• torch
• torchvision

Installation

Using pip

If desired, create a conda environment. To install the required dependencies, run the following command:

pip install timm torchvision matplotlib

Training

To train the model, modify the directories in maskrcnn.py to where your dataset is loaded. Then run:

python maskrcnn.py