mapreader 1.8.2__tar.gz → 1.8.2.post0.dev15__tar.gz

{mapreader-1.8.2 → mapreader-1.8.2.post0.dev15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mapreader
-Version: 1.8.2
+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 → mapreader-1.8.2.post0.dev15}/mapreader/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2025-12-19T16:13:11+0000",
+ "date": "2026-03-25T21:06:12+0000",
  "dirty": false,
  "error": null,
- "full-revisionid": "a4d302f15a8d7d2ee06cfc30c86a5c2a735ba858",
- "version": "1.8.2"
+ "full-revisionid": "08fceec9e3feef2a5c60385dba516f7639897f03",
+ "version": "1.8.2.post0.dev15"
 }
 '''  # END VERSION_JSON
 

{mapreader-1.8.2 → mapreader-1.8.2.post0.dev15}/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
@@ -565,7 +590,7 @@ Use ``torch.optim.lr_scheduler`` directly and then the ``add_scheduler`` method
             col_names = ["output_size", "output_size", "num_params"]
 
         model_summary = summary(
-            self.model, input_size=input_size, col_names=col_names, **kwargs
+            self.model, input_size=input_size, col_names=col_names, device=self.device, **kwargs
         )
         print(model_summary)
 

{mapreader-1.8.2 → mapreader-1.8.2.post0.dev15}/mapreader.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mapreader
-Version: 1.8.2
+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