Skip to content

Commit

Permalink
updated tags
Browse files Browse the repository at this point in the history
  • Loading branch information
mhubii committed Mar 30, 2022
1 parent 5f1a61a commit 2eab6a0
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions endoscopy/utils/loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,11 +8,11 @@ class MODEL(object):
URLs to models are to include a 'cpu' or 'cuda' tag, respectively.
"""
class HOMOGRAPHY_ESTIMATION_ENUM(Enum):
RESNET_34 = "https://github.com/RViMLab/endoscopy/releases/download/0.0.1/h_est_resnet_34_{}.pt"
RESNET_34 = "https://github.com/RViMLab/endoscopy/releases/download/0.1.1/h_est_resnet_34_{}.pt"

class SEGMENTATION_ENUM(Enum):
UNET_RESNET_34 = "https://github.com/RViMLab/endoscopy/releases/download/0.0.1/seg_unet_resnet_34_{}.pt"
UNET_RESNET_34_TINY = "https://github.com/RViMLab/endoscopy/releases/download/0.0.1/seg_unet_resnet_34_{}_tiny.pt"
UNET_RESNET_34 = "https://github.com/RViMLab/endoscopy/releases/download/0.1.1/seg_unet_resnet_34_{}.pt"
UNET_RESNET_34_TINY = "https://github.com/RViMLab/endoscopy/releases/download/0.1.1/seg_unet_resnet_34_{}_tiny.pt"

HOMOGRAPHY_ESTIMATION = HOMOGRAPHY_ESTIMATION_ENUM
SEGMENTATION = SEGMENTATION_ENUM
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@

setuptools.setup(
name="endoscopy",
version="0.0.14",
version="0.1.1",
author="Martin Huber",
author_email="[email protected]",
description="Image processing utilities for endoscopic images.",
Expand Down

0 comments on commit 2eab6a0

Please sign in to comment.