mapreader 1.8.2.post0.dev9__py3-none-any.whl → 1.8.2.post0.dev15__py3-none-any.whl

mapreader/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2026-02-26T15:15:54+0000",
+ "date": "2026-03-25T21:06:12+0000",
  "dirty": false,
  "error": null,
- "full-revisionid": "87578a471ab1dcadf80a2552aa31a3f50cac862a",
- "version": "1.8.2.post0.dev9"
+ "full-revisionid": "08fceec9e3feef2a5c60385dba516f7639897f03",
+ "version": "1.8.2.post0.dev15"
 }
 '''  # END VERSION_JSON
 

mapreader/classify/classifier.py

@@ -108,6 +108,7 @@ class ClassifierContainer:
         is_inception: bool = False,
         load_path: str | None = None,
         force_device: bool = False,
+        huggingface: bool = False,
         **kwargs,
     ):
         # set up device
@@ -149,7 +150,31 @@ class ClassifierContainer:
                 self.input_size = input_size
                 self.is_inception = is_inception
             elif isinstance(model, str):
-                self._initialize_model(model, **kwargs)
+                if huggingface:
+                    try:
+                        from transformers import AutoModelForImageClassification, AutoImageProcessor
+                    except ImportError:
+                        raise ImportError(
+                            "Hugging Face models require the 'transformers' library: 'pip install transformers'."
+                        )
+                    print(f"[INFO] Initializing Hugging Face model: {model}")
+                    num_labels = len(self.labels_map)
+                    self.model = AutoModelForImageClassification.from_pretrained(
+                        model,
+                        num_labels=num_labels,
+                        ignore_mismatched_sizes=True
+                    ).to(self.device)
+                    hf_processor = AutoImageProcessor.from_pretrained(model)
+                    size = getattr(hf_processor, "size", {})
+                    if "height" in size and "width" in size:
+                        size = (size["height"], size["width"])
+                    elif "shortest_edge" in size:
+                        size = (size["shortest_edge"], size["shortest_edge"])
+                    else:
+                        size = input_size
+                    self.is_inception = False
+                else:
+                    self._initialize_model(model, **kwargs)
 
             self.optimizer = None
             self.scheduler = None

{mapreader-1.8.2.post0.dev9.dist-info → mapreader-1.8.2.post0.dev15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mapreader
-Version: 1.8.2.post0.dev9
+Version: 1.8.2.post0.dev15
 Summary: A computer vision pipeline for the semantic exploration of maps/images at scale
 Home-page: https://github.com/Living-with-machines/MapReader
 Download-URL: https://github.com/Living-with-machines/MapReader/archive/refs/heads/main.zip

{mapreader-1.8.2.post0.dev9.dist-info → mapreader-1.8.2.post0.dev15.dist-info}/RECORD

@@ -1,10 +1,10 @@
 mapreader/__init__.py,sha256=ztmWx0OCneJzlu2L1CP7NPwieGYv1kvARJJOUzq_6OQ,1720
 mapreader/__main__.py,sha256=hKOJEZkRPHZ-wwwz_RRR5OxMK5UilAfbQlq6k9icl_I,112
-mapreader/_version.py,sha256=ojxsMkAf2NCXx9VAS0VvM2OgHg_yI_oq5FRlU_gaz78,508
+mapreader/_version.py,sha256=j0h4Cm-RpugAeKh-aYpb1J_eILj79Su2Bk1LQUjZ6iA,509
 mapreader/annotate/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mapreader/annotate/annotator.py,sha256=x2K8yn2WXYOMTajgSQ-iXff4cIzUQHaYJUpw0S7AIzI,37508
 mapreader/classify/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mapreader/classify/classifier.py,sha256=P--3keloxsHSMUhHCquLTFfEMO-t0LKWUu5k34fr5bY,71173
+mapreader/classify/classifier.py,sha256=NIJ-VFWr0YW6Nug7hWovcmCbiEEw-tZcdNbmo02IBDU,72502
 mapreader/classify/custom_models.py,sha256=1qwlpUjIIdD-syCt7CA-4HSNUfVAgISR-MPQI7taA-w,2170
 mapreader/classify/datasets.py,sha256=stwIndSOomU5yFl_Ay2m6e2QYO0XHrtRWaUWdNCKcXg,30442
 mapreader/classify/load_annotations.py,sha256=kVvyVkBSzmEfRXV5hM8dN-EkP9cgu5bR6kS5Chipf1I,34887
@@ -34,9 +34,9 @@ mapreader/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mapreader/utils/compute_and_save_stats.py,sha256=bOI6FfnzLWwtx8YOY-A7xN1mK1Thx7De-FANGJ2BNRY,979
 mapreader/utils/load_frames.py,sha256=o4QztKbf8FBs6cxDHtx-ypwIMZRf4px6q1aKD5kZLwI,3558
 mapreader/utils/slice_parallel.py,sha256=jXHYZ811IiKBPOKPI40IYeSA9RHC7evo61H44fbPXhI,1634
-mapreader-1.8.2.post0.dev9.dist-info/licenses/LICENSE,sha256=lZ12qYK_-T-58n5YueZUgyxZQpz4kJzGCJwAI9e2rXA,1215
-mapreader-1.8.2.post0.dev9.dist-info/METADATA,sha256=yWDDXJrft2aCJIzvF10NClUj3yOqPONE-eoioQ0XCMQ,15833
-mapreader-1.8.2.post0.dev9.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
-mapreader-1.8.2.post0.dev9.dist-info/entry_points.txt,sha256=oJAMoNzUMr-iGOli2-o2hlzYbyPPM_eVHQDFrijQvz8,54
-mapreader-1.8.2.post0.dev9.dist-info/top_level.txt,sha256=Z1DD0SVhh767afeDuV7blzVA6sYl0-71K81-ld4EbKY,10
-mapreader-1.8.2.post0.dev9.dist-info/RECORD,,
+mapreader-1.8.2.post0.dev15.dist-info/licenses/LICENSE,sha256=lZ12qYK_-T-58n5YueZUgyxZQpz4kJzGCJwAI9e2rXA,1215
+mapreader-1.8.2.post0.dev15.dist-info/METADATA,sha256=4O1ltwJEC6nfAa5bjUVyP5jmi4hnFFIUZZJP_Yc6P8s,15834
+mapreader-1.8.2.post0.dev15.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+mapreader-1.8.2.post0.dev15.dist-info/entry_points.txt,sha256=oJAMoNzUMr-iGOli2-o2hlzYbyPPM_eVHQDFrijQvz8,54
+mapreader-1.8.2.post0.dev15.dist-info/top_level.txt,sha256=Z1DD0SVhh767afeDuV7blzVA6sYl0-71K81-ld4EbKY,10
+mapreader-1.8.2.post0.dev15.dist-info/RECORD,,

{mapreader-1.8.2.post0.dev9.dist-info → mapreader-1.8.2.post0.dev15.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (82.0.0)
+Generator: setuptools (82.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any