https://github.com/bowang-lab/maester

Masked Autoencoder Guided Segmentation at Pixel Resolution

https://github.com/bowang-lab/maester

Science Score: 36.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • CITATION.cff file
  • codemeta.json file
    Found codemeta.json file
  • .zenodo.json file
    Found .zenodo.json file
  • DOI references
  • Academic publication links
  • Committers with academic emails
    1 of 2 committers (50.0%) from academic institutions
  • Institutional organization owner
  • JOSS paper metadata
  • Scientific vocabulary similarity
    Low similarity (8.4%) to scientific vocabulary
Last synced: 10 months ago · JSON representation

Repository

Masked Autoencoder Guided Segmentation at Pixel Resolution

Basic Info
  • Host: GitHub
  • Owner: bowang-lab
  • License: mit
  • Language: Python
  • Default Branch: main
  • Size: 1.39 MB
Statistics
  • Stars: 15
  • Watchers: 4
  • Forks: 6
  • Open Issues: 2
  • Releases: 0
Created over 3 years ago · Last pushed almost 3 years ago
Metadata Files
Readme License

README.md

CVPR2023 Highlight | MAESTER: Masked Autoencoder Guided Segmentation at Pixel Resolution for Accurate, Self-Supervised Subcellular Structure Recognition

Check out the Paper! and our Youtube Talk!

💥 Introduction

We introduce MAESTER (Masked AutoEncoder guided SegmenTation at pixEl Resolution), a self-supervised method for accurate, subcellular structure segmentation at pixel resolution. MAESTER treats volume electron microscopy(vEM) image segmentation as a representation learning and clustering problem. Specifically, MAESTER learns semantically meaningful token representations of multi-pixel image patches while simultaneously maintaining a sufficiently large field of view for contextual learning. We also develop a cover-and-stride inference strategy to achieve pixel-level subcellular strueture segmentation.

⚙️ Installation

  • Clone the repository:

git clone https://github.com/bowang-lab/MAESTER

  • Set up the environment:

poetry install poetry shell pip install torch==2.0.1 torchvision==0.15.2

  • Download the trained model for demo

    • Google drive: https://drive.google.com/drive/folders/143W_VSl5ONE3NGbnI0i19S8lBRml7lRz?usp=sharing
    • Put it under ./MAESTER/model_weights/
  • Dataset:

wget https://cloud.mpi-cbg.de/index.php/s/UJopHTRuh6f4wR8/download

  • and unzip the dataset, put it under ./MAESTER/data/

🎉 Example

  • Check out our detailed demo:
    • Inference with MAESTER ./examples/inference_demo.ipynb.

📝 To-do

  • [x] Add inference demo
  • [ ] Add scalable inference example
  • [ ] Add DDP training example

📄 Citation

@InProceedings{Xie_2023_CVPR, author = {Xie, Ronald and Pang, Kuan and Bader, Gary D. and Wang, Bo}, title = {MAESTER: Masked Autoencoder Guided Segmentation at Pixel Resolution for Accurate, Self-Supervised Subcellular Structure Recognition}, booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)}, month = {June}, year = {2023}, pages = {3292-3301} }

Acknowledgement

  • This repository is built upon MAE.

Owner

  • Name: WangLab @ U of T
  • Login: bowang-lab
  • Kind: organization
  • Location: 190 Elizabeth St, Toronto, ON M5G 2C4 Canada

BoWang's Lab at University of Toronto

GitHub Events

Total
  • Watch event: 2
  • Fork event: 1
Last Year
  • Watch event: 2
  • Fork event: 1

Committers

Last synced: about 1 year ago

All Time
  • Total Commits: 2
  • Total Committers: 2
  • Avg Commits per committer: 1.0
  • Development Distribution Score (DDS): 0.5
Past Year
  • Commits: 0
  • Committers: 0
  • Avg Commits per committer: 0.0
  • Development Distribution Score (DDS): 0.0
Top Committers
Name Email Commits
Ronald Xie 4****n 1
Kuan-Pang k****g@m****a 1
Committer Domains (Top 20 + Academic)

Issues and Pull Requests

Last synced: about 1 year ago

All Time
  • Total issues: 1
  • Total pull requests: 1
  • Average time to close issues: N/A
  • Average time to close pull requests: N/A
  • Total issue authors: 1
  • Total pull request authors: 1
  • Average comments per issue: 1.0
  • Average comments per pull request: 0.0
  • Merged pull requests: 0
  • Bot issues: 0
  • Bot pull requests: 0
Past Year
  • Issues: 0
  • Pull requests: 0
  • Average time to close issues: N/A
  • Average time to close pull requests: N/A
  • Issue authors: 0
  • Pull request authors: 0
  • Average comments per issue: 0
  • Average comments per pull request: 0
  • Merged pull requests: 0
  • Bot issues: 0
  • Bot pull requests: 0
Top Authors
Issue Authors
  • AlexSauer (1)
Pull Request Authors
  • mese79 (1)
Top Labels
Issue Labels
Pull Request Labels