Error if I want to pass a vector file path as input for predict()

[ericko443]

Environment Information

• samgeo version: 0.10.6
• Python version: 3.12.3
• Operating System: Ubuntu 20.04.6 LTS

Description

I wanted to run sam.predict() on an image for which I previously generated a vector file (.shp) containing several point geometries indicating the objects I want to segment. The documentation says that the points to be used in sam.predict() can also be specified as a file path to a vector dataset. However, I keep receiving an error (see below). I think the error lies in the vector_to_geojson() function that gets called within predict(). If I don't supply it with a URL then it fails because it wants to call download_file() but I only specify a file path so I would not expect download_file() to be called. Is this a bug? Or did I misunderstand the allowed input format?

What I Did

Code:

from samgeo import SamGeo

checkpoint = '.../sam_vit_h_4b8939.pth'
image = '.../input_img.tif'
randpoints = '.../input_pts.shp'

device = 'cuda' if torch.cuda.is_available() else 'cpu'
sam = SamGeo(
    checkpoint=checkpoint,
    model_type='vit_h',
    device=device,
    sam_kwargs=None,
    automatic=False,
)

sam.set_image(image)
sam.predict(randpoints)

Output:

Traceback (most recent call last):

  Cell In[45], line 17
    sam.predict(randpoints)

  File ~/anaconda3/envs/spyder-env/lib/python3.12/site-packages/samgeo/samgeo.py:533 in predict
    point_coords = vector_to_geojson(point_coords)

  File ~/anaconda3/envs/spyder-env/lib/python3.12/site-packages/samgeo/common.py:727 in vector_to_geojson
    filename = download_file(filename)

  File ~/anaconda3/envs/spyder-env/lib/python3.12/site-packages/samgeo/common.py:146 in download_file
    out_dir = os.path.abspath(os.path.dirname(output))

  File <frozen posixpath>:181 in dirname

TypeError: expected str, bytes or os.PathLike object, not NoneType