Skip to content

An image segmentation app that combines the power of YoloV8 and Segment Anything model (SAM)

Notifications You must be signed in to change notification settings

roydipta/safari_stickerbook

Repository files navigation

Safari Stickerbook

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
Demo GIF

Steps for Installation:

 Prereq: Python 3.9.16 is required.

  1. From your terminal clone this repository by using: git clone https://github.com/roydipta/capstone.git
  2. Navigate to the directory where downloaded
  3. Create a virtual environment python3 -m venv name_of_environment
  4. Activate the virtual environment:
      on macOS or Linux use: source name_of_environment/bin/activate
      on windows use: .\name_of_environment\Scripts\activate
  5. Now pip install the requirement.txt file by using the command: pip install -r requirements.txt
  6. 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.

Citations

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 image

About

An image segmentation app that combines the power of YoloV8 and Segment Anything model (SAM)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages