From 7bd9601ce55a7cf6a6506362d5d804fdb1a8e78b Mon Sep 17 00:00:00 2001 From: Kohulan Rajan Date: Thu, 15 Feb 2024 09:58:46 +0100 Subject: [PATCH] fix: decimer-segmentation improved model --- decimer_segmentation/decimer_segmentation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/decimer_segmentation/decimer_segmentation.py b/decimer_segmentation/decimer_segmentation.py index ba3c030..c899b7e 100644 --- a/decimer_segmentation/decimer_segmentation.py +++ b/decimer_segmentation/decimer_segmentation.py @@ -173,7 +173,7 @@ def load_model() -> modellib.MaskRCNN: # Download trained weights if needed if not os.path.exists(model_path): print("Downloading model weights...") - url = "https://zenodo.org/record/7228583/files/mask_rcnn_molecule.h5?download=1" + url = "https://zenodo.org/record/10663579/files/mask_rcnn_molecule.h5?download=1" req = requests.get(url, allow_redirects=True) with open(model_path, "wb") as model_file: model_file.write(req.content)