Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add support for SAM 2 #312

Merged
merged 16 commits into from
Sep 16, 2024
Merged

Add support for SAM 2 #312

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
2e42d5e
Add samgeo2 module
giswqs Sep 12, 2024
3590d05
Add SamGeo2 class
giswqs Sep 13, 2024
916e5aa
Add mask and anns functions
giswqs Sep 13, 2024
f0d2e7c
Add predidct function
giswqs Sep 14, 2024
7040d0e
Add video predictor
giswqs Sep 14, 2024
804de1b
Clean up code
giswqs Sep 14, 2024
d3cfb13
Add map gui
giswqs Sep 14, 2024
b07ffc7
Add images and video conversion
giswqs Sep 14, 2024
7658ea5
Add predict_video function
giswqs Sep 15, 2024
b37bfed
Add show_image_gui function
giswqs Sep 15, 2024
c38d149
Save prediction result
giswqs Sep 15, 2024
e4b815f
Add save video prediction blended
giswqs Sep 15, 2024
7e5c754
Add show_prompts and show_images functions
giswqs Sep 16, 2024
1a35669
Add support for GeoTIFF
giswqs Sep 16, 2024
404fde0
Improve images_to_video function
giswqs Sep 16, 2024
78567d6
Add support point crs
giswqs Sep 16, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
[![image](https://img.shields.io/conda/vn/conda-forge/segment-geospatial.svg)](https://anaconda.org/conda-forge/segment-geospatial)
[![Docker Pulls](https://badgen.net/docker/pulls/giswqs/segment-geospatial?icon=docker&label=pulls)](https://hub.docker.com/r/giswqs/segment-geospatial)
[![PyPI Downloads](https://static.pepy.tech/badge/segment-geospatial)](https://pepy.tech/project/segment-geospatial)
[![Conda Recipe](https://img.shields.io/badge/recipe-segment--geospatial-green.svg)](https://anaconda.org/conda-forge/segment-geospatial)
[![Conda Downloads](https://anaconda.org/conda-forge/segment-geospatial/badges/downloads.svg)](https://anaconda.org/conda-forge/segment-geospatial)
[![DOI](https://joss.theoj.org/papers/10.21105/joss.05663/status.svg)](https://doi.org/10.21105/joss.05663)

Expand Down
1 change: 1 addition & 0 deletions docs/index.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
[![image](https://img.shields.io/conda/vn/conda-forge/segment-geospatial.svg)](https://anaconda.org/conda-forge/segment-geospatial)
[![Docker Pulls](https://badgen.net/docker/pulls/giswqs/segment-geospatial?icon=docker&label=pulls)](https://hub.docker.com/r/giswqs/segment-geospatial)
[![PyPI Downloads](https://static.pepy.tech/badge/segment-geospatial)](https://pepy.tech/project/segment-geospatial)
[![Conda Recipe](https://img.shields.io/badge/recipe-segment--geospatial-green.svg)](https://anaconda.org/conda-forge/segment-geospatial)
[![Conda Downloads](https://anaconda.org/conda-forge/segment-geospatial/badges/downloads.svg)](https://anaconda.org/conda-forge/segment-geospatial)
[![DOI](https://joss.theoj.org/papers/10.21105/joss.05663/status.svg)](https://doi.org/10.21105/joss.05663)

Expand Down
3 changes: 3 additions & 0 deletions docs/samgeo2.md
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# samgeo2 module

::: samgeo.samgeo2
1 change: 1 addition & 0 deletions mkdocs.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,7 @@ nav:
- API Reference:
- common module: common.md
- samgeo module: samgeo.md
- samgeo2 module: samgeo2.md
- fast_sam module: fast_sam.md
- hq_sam module: hq_sam.md
- text_sam module: text_sam.md
1 change: 1 addition & 0 deletions requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
gdown
geopandas
huggingface_hub
ipympl
leafmap
localtileserver
matplotlib
Expand Down
2 changes: 2 additions & 0 deletions samgeo/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,3 +6,5 @@


from .samgeo import *
from .samgeo2 import *
from .common import show_image_gui
Loading
Loading