The project is for class CMSC426 Computer Vision.
Assignment Due Date: May 9, 2023
Assignment Score: 43 out of 50
The language used: Jupyter Notebook
Assignment Purpose: This assignment used PyTorch to train neural networks for the tasks of image classification and image segmentation.
Some Processing Outcomes:
Functions implemented:
- Training an Image Classifier
- Define a Convolutional Neural Network
- Define a Loss function and optimizer
- Train the network
- Test the network on the test data
- Semantic Segmentation
def __init__(self, dir, color_map)def convert_to_segmentation_mask(self,mask)def __getitem__(self,index)