Skip to content

Here’s an initial version of a web-based macroalgae auto-labeling tool using Streamlit.πŸ”§ What This App Does: Lets you upload underwater macroalgae images Applies your custom YOLO + SAM models to segment algae Displays: The original image Overlay with segmentation masks Class-wise percentage coverage

DolapoSalim/web-based-macroalgae-active-learner

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

21 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

Macroalgae Auto-Label + Coverage Web Tools

🧩 Core Features
Features Description
πŸ“€ Upload new underwater images User can upload new image(s) via the web interface
βš™οΈ Auto-label with YOLO + SAM YOLO detects β†’ SAM segments each detection
πŸ–ΌοΈ Display results interactively Show segmentation overlays + class-wise masks
πŸ“Š Calculate % coverage Pixel-based area per class (for each image)
πŸ’Ύ Download annotations & coverage Export JSON or image+mask zip archive
πŸ› οΈ Tech Stack
  • Frontend + Backend: Streamlit (modern, fast, interactive UI)
  • Model Backend: Your fine-tuned YOLO model + pretrained SAM model
  • Image Ops: OpenCV & NumPy
  • Optional Storage: Save results locally or integrate with Firebase/GCP later.
πŸ“ Directory Structure
project/
β”œβ”€β”€ custom-yolo/           # folder containing pre-trained YOLO model but .gitignored
β”œβ”€β”€ images/                # raw images to label
β”œβ”€β”€ output/
β”‚   β”œβ”€β”€ masks/             # generated masks
β”‚   β”œβ”€β”€ overlays/          # visualization of predictions
β”‚   └── annotations.json   # structured result
β”œβ”€β”€ sam-model              # contains the Segment Anything Model
└── scr/                   # contains notebook
|   β”œβ”€β”€ app.py
β”œβ”€β”€ readMe.md
β”œβ”€β”€ requirements.txt
πŸ“‚ To Run It Locally:
  • Save this file as app.py
  • Place yolo_model.pt and sam_b.pt in the same directory
  • Run the requirement file
pip install -r requirements.txt
  • Install dependencies:
pip install streamlit opencv-python pillow ultralytics
  • Run the app:
streamlit run project/scr/app.py

About

Here’s an initial version of a web-based macroalgae auto-labeling tool using Streamlit.πŸ”§ What This App Does: Lets you upload underwater macroalgae images Applies your custom YOLO + SAM models to segment algae Displays: The original image Overlay with segmentation masks Class-wise percentage coverage

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published