| --- |
| license: mit |
| datasets: |
| - cvdl/oxford-pets |
| tags: |
| - code |
| --- |
| |
| # Math 443 Final Project Models |
| This project contains image learning models for group 4's final project for UW-Madison's Math 443: Applied Linear Algebra. |
| There are two main directories in this project, `oxford-pets` and `cityscapes`. |
| Each repository contains the models trained on those datasets respectively. |
|
|
| # Oxford Pets |
| `oxford-pets` contains two models for the dataset, the image classifier and image masker. |
| The ensemble model of these two is roughly a single-object segmentation model. |
| These were trained from random weights, and gave decent results. |
|
|
| # CityScapes |
| `cityscapes` contains a single model for making instance segmentation predictions on the CityScapes images. |
| The model was fine-tuned from the torchvision Mask R-CNN w/ ResNet50 backbone. |
