This app combines the classifcation power of YOLOV8 and the segmentation capabilities of Meta's Segment Anything Model (SAM) to create a easy to use animal segmentation app.
Webpage live on: https://my-flask-app-ftuis4n5rq-ue.a.run.app
Prereq: Python 3.9.16 is required.
- From your terminal clone this repository by using:
git clone https://github.com/roydipta/capstone.git
- Navigate to the directory where downloaded
- Create a virtual environment
python3 -m venv name_of_environment
- Activate the virtual environment:
on macOS or Linux use:source name_of_environment/bin/activate
on windows use:.\name_of_environment\Scripts\activate
- Now pip install the requirement.txt file by using the command:
pip install -r requirements.txt
- Lastly use
flask run
to run the application
This will also download the model weights file (about 350 mb) for you on your local drive where the repo is cloned.
Kirillov, A., Mintun, E., Ravi, N., Mao, H., Rolland, C., Gustafson, L., Xiao, T., Whitehead, S., Berg, A. C., Lo, W.-Y., Dollár, P., & Girshick, R. (2023). Segment Anything. arXiv:2304.02643 [cs.CV]. https://doi.org/10.48550/arXiv.2304.02643