Comments (4)
Is this related to [https://github.com//issues/201] ?
from segment-geospatial.
No. Not related to #201. I want to run samgeo.hq_sam on the CPU in the same way you can run "normal" samgeo on the CPU.
The solution proposed in #173 does not work, whether you install locally or in a docker container.
The speed difference between running inference with samgeo on a GPU or a CPU is not too dramatic, esp. when testing (GPU order of 6 faster). The models require a pretty large GPU forcing you to move to cloud resources. I prefer running on CPU.
from segment-geospatial.
Try adding device='cpu'
when initializing SamGeo:
from samgeo.hq_sam import SamGeo
sam = SamGeo(model_type="vit_h", sam_kwargs=None, device='cpu')
See if it helps!
from segment-geospatial.
Thanks for the tip, but I tried that, and it doesn't work. The problem is in the sam_hq implementation, which does not check for the device key. To get this solved, you need to go into /opt/conda/lib/python3.11/site-packages/segment_anything_hq/build_sam.py
and edit line 160:
state_dict = torch.load(f, map_location=torch.device('cpu'))
which is for the vit_h model, you need to edit the equivalent statements for the other models. This is the quick and dirty solution, it would be better to do this on the basis of the device keyword.
(I assume use of the giswqs/segment-geospatial docker image. I (locally) commit the revised container to giswqs/segment-geospatial-cpu)
from segment-geospatial.
Related Issues (20)
- self.save_masks unique value mask results are covered HOT 1
- Initialise FastSAM is not working HOT 7
- samgeo installation failed on arcggis pro 2.9 HOT 2
- Problem with the Interactive segmentation using Points on Image HOT 1
- Arcgis Pro 3.2 samgeo installation error: rumtime error HOT 1
- batch predict
- sam.generate(): Error with uploaded images HOT 2
- warning libmamba Added empty dependency for problem type SOLVER_RULE_UPDATE Could not solve for environment specs The following package could not be installed └─ segment-geospatial is installable and it requires └─ pycocotools with the potential options HOT 5
- SamGeo() cannot download model vit_h (samgeo.hq_sam) HOT 7
- Expose leafmap backend options to user HOT 7
- exception when using box prompt
- ImportError: DLL load failed while importing etree: The specified procedure could not be found. HOT 1
- Add support for Clay Foundation Model
- ArcPro 3.2.2 Install Error HOT 1
- Is using headless opencv an option? HOT 1
- Fine Tuning HQ SAM HOT 1
- [Dependency] ModuleNotFoundError: No module named 'xarray'
- Failed to visulize the map HOT 2
- Assistance Needed with ArcGIS Pro 3.3 HOT 4
- More objects in annotations than in mask!
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from segment-geospatial.