Skip to content

Commit 4a3cbde

Browse files
committed
Update AI model URLs
1 parent e5da29c commit 4a3cbde

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

stages/dataset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -221,7 +221,7 @@ def find_mask(photo_path, masks):
221221

222222
if len(sky_images) > 0:
223223
log.ODM_INFO("Automatically generating sky masks for %s images" % len(sky_images))
224-
model = ai.get_model("skyremoval", "https://github.com/OpenDroneMap/SkyRemoval/releases/download/v1.0.5/model.zip", "v1.0.5")
224+
model = ai.get_model("skyremoval", "https://github.com/WebODM/ODX/releases/download/v3.7.1/skyremoval.zip", "v1.0.5")
225225
if model is not None:
226226
sf = SkyFilter(model=model)
227227

@@ -262,7 +262,7 @@ def parallel_sky_filter(item):
262262

263263
if len(bg_images) > 0:
264264
log.ODM_INFO("Automatically generating background masks for %s images" % len(bg_images))
265-
model = ai.get_model("bgremoval", "https://github.com/OpenDroneMap/ODM/releases/download/v2.9.0/u2net.zip", "v2.9.0")
265+
model = ai.get_model("bgremoval", "https://github.com/WebODM/ODX/releases/download/v3.7.1/u2net.zip", "v2.9.0")
266266
if model is not None:
267267
bg = BgFilter(model=model)
268268

0 commit comments

Comments
 (0)