Deep Learning Research - COMP 499
Drew Davinack (PhD), Mark LeBlanc (PhD), Avery Chan, Paedar Rader, Sayed Ibrahimi
This project is a study of the health of oysters, specifically the infestation of mud blisters from burrowing worms, to build an image recognition model that can accurately predict how much surface area of an oyster is infected with the parasites. This section is a work in progress.
Make sure the following are installed on local machine/cloud device:
- Lastest version of Anaconda (Python 3.12)
- PyTorch
- Install with Conda:
- Install with pip:
conda install pytorch torchvision -c pytorchpip3 install torch torchvision - Skorch (NN Dependency)
- Install with Conda:
- Install with pip:
- Latest version of Python (3.12)
- This section is subject to changes
git clone https://github.com/skorch-dev/skorch.git
cd skorch
conda create -n skorch-env python=3.10
conda activate skorch-env
# install pytorch version for your system (see below)
python -m pip install -r requirements.txt
python -m pip install .
python -m pip install -U skorch
-- Resnet-50 --
@article{He2015,
author = {Kaiming He and Xiangyu Zhang and Shaoqing Ren and Jian Sun},
title = {Deep Residual Learning for Image Recognition},
journal = {arXiv preprint arXiv:1512.03385},
year = {2015}
}
- Corresponding Repo: *
