Upload ResNet-18 genus model (epoch 3)

README.md

@@ -0,0 +1,76 @@
+---
+license: mit
+library_name: deepforest
+pipeline_tag: image-classification
+tags:
+  - deepforest
+  - crop-model
+  - tree-genus
+  - ecology
+  - neon
+---
+
+# Tree Genus Classification (CropModel)
+
+Classifies tree crowns detected by [DeepForest](https://github.com/weecology/DeepForest) into 54 genera. Trained on RGB imagery from 29 [NEON](https://www.neonscience.org/) sites across North America.
+
+Trained with [NeonTreeClassification](https://github.com/GatorSense/NeonTreeClassification).
+
+## Usage
+
+```python
+from deepforest import main
+from deepforest.model import CropModel
+
+detector = main.deepforest()
+detector.load_model("weecology/deepforest-tree")
+
+genus_model = CropModel.load_model("weecology/cropmodel-tree-genus")
+
+results = detector.predict_tile(path="tile.tif", crop_model=genus_model)
+# results has columns: cropmodel_label, cropmodel_score
+```
+
+## Results (Test Set)
+
+| Metric | Value |
+|---|---|
+| Accuracy | 44.0% |
+| Macro F1 | 0.25 |
+| Weighted F1 | 0.44 |
+| Classes | 54 |
+
+Full per-class precision/recall/F1 in [`classification_report.csv`](classification_report.csv).
+
+## Training
+
+| Parameter | Value |
+|---|---|
+| Architecture | ResNet-18 (torchvision, ImageNet pretrained) |
+| Input | 224x224 RGB, ImageNet normalization |
+| Resize interpolation | nearest-neighbor |
+| Optimizer | AdamW (lr=2.5e-4, weight_decay=1e-4) |
+| Scheduler | ReduceLROnPlateau |
+| Max epochs | 500 (early stopping patience=15) |
+| Best epoch | 3 (val_loss=2.22) |
+| Batch size | 256 |
+| Class weights | sqrt inverse-frequency |
+| Seed | 42 |
+
+## Dataset
+
+16,348 deduplicated tree crowns from 29 NEON sites. One sample per unique individual, rare species (<6 samples) removed. Labels from NEON Vegetation Structure Taxonomy (VST) field surveys. RGB crown crops extracted at 0.1m resolution.
+
+| Split | Samples |
+|---|---|
+| Train (70%) | 11,443 |
+| Val (15%) | 2,452 |
+| Test (15%) | 2,453 |
+
+Split method: stratified random, seed=42.
+
+**Sites**: ABBY, BART, BONA, CLBJ, DEJU, DELA, GRSM, GUAN, HARV, HEAL, JERC, KONZ, LENO, MLBS, MOAB, NIWO, ONAQ, OSBS, PUUM, RMNP, SCBI, SERC, SJER, SOAP, TALL, TEAK, UKFS, UNDE, WREF
+
+## License
+
+MIT

classification_report.csv

@@ -0,0 +1,58 @@
+,precision,recall,f1-score,support
+Abies (0),0.3672,0.4159,0.39,113.0
+Acer (1),0.3586,0.4424,0.3961,278.0
+Alnus (2),0.0,0.0,0.0,4.0
+Amelanchier (3),0.125,0.2632,0.1695,19.0
+Arctostaphylos (4),0.0,0.0,0.0,1.0
+Betula (5),0.641,0.2525,0.3623,99.0
+Bourreria (6),1.0,1.0,1.0,3.0
+Bucida (7),1.0,0.5,0.6667,2.0
+Bursera (8),1.0,1.0,1.0,2.0
+Calocedrus (9),0.4667,0.6667,0.549,21.0
+Carpinus (10),0.0909,0.0333,0.0488,30.0
+Carya (11),0.2329,0.3953,0.2931,43.0
+Celtis (12),0.1538,0.0426,0.0667,47.0
+Cercis (13),0.3,0.5,0.375,6.0
+Cornus (14),0.1111,0.1667,0.1333,6.0
+Diospyros (15),0.0,0.0,0.0,1.0
+Fagus (16),0.2621,0.5094,0.3462,53.0
+Fraxinus (17),0.2326,0.2,0.2151,50.0
+Gleditsia (18),0.0,0.0,0.0,3.0
+Gymnocladus (19),0.0,0.0,0.0,1.0
+Halesia (20),0.0,0.0,0.0,3.0
+Ilex (21),0.0,0.0,0.0,1.0
+Juglans (22),0.2632,0.4545,0.3333,11.0
+Juniperus (23),0.2162,0.4211,0.2857,19.0
+Larix (24),0.0,0.0,0.0,4.0
+Liquidambar (25),0.2407,0.2321,0.2364,56.0
+Liriodendron (26),0.3448,0.5085,0.411,59.0
+Maclura (27),0.1111,0.1429,0.125,7.0
+Magnolia (28),0.1667,0.3333,0.2222,6.0
+Metrosideros (29),1.0,0.7143,0.8333,7.0
+Morus (30),0.0,0.0,0.0,6.0
+Nyssa (31),0.0,0.0,0.0,26.0
+Ostrya (32),0.0,0.0,0.0,1.0
+Oxydendrum (33),0.1579,0.1579,0.1579,19.0
+Picea (34),0.7692,0.65,0.7046,200.0
+Pinaceae (35),0.0,0.0,0.0,2.0
+Pinus (36),0.6012,0.6293,0.6149,321.0
+Pisonia (37),0.6667,0.6667,0.6667,3.0
+Platanus (38),0.5,0.1429,0.2222,7.0
+Populus (39),0.6173,0.5814,0.5988,86.0
+Prunus (40),0.0,0.0,0.0,9.0
+Pseudotsuga (41),0.5806,0.63,0.6043,200.0
+Quercus (42),0.5315,0.3665,0.4338,322.0
+Robinia (43),0.0,0.0,0.0,5.0
+Sassafras (44),0.5,0.1667,0.25,6.0
+Sideroxylon (45),0.0,0.0,0.0,1.0
+Symphoricarpos (46),0.0,0.0,0.0,2.0
+Taxus (47),0.0,0.0,0.0,4.0
+Thuja (48),0.0,0.0,0.0,6.0
+Tilia (49),0.0,0.0,0.0,2.0
+Triadica (50),0.0,0.0,0.0,2.0
+Tsuga (51),0.4,0.3889,0.3944,216.0
+Ulmus (52),0.1818,0.4615,0.2609,39.0
+Unknown (53),0.0,0.0,0.0,13.0
+accuracy,0.4403,0.4403,0.4403,0.4403
+macro avg,0.2628,0.2599,0.2475,2453.0
+weighted avg,0.454,0.4403,0.4358,2453.0

config.json

@@ -0,0 +1,84 @@
+{
+  "cropmodel": {
+    "architecture": "resnet18",
+    "batch_size": 4,
+    "num_workers": 0,
+    "lr": 0.0001,
+    "scheduler": {
+      "type": "ReduceLROnPlateau",
+      "params": {
+        "mode": "min",
+        "factor": 0.5,
+        "patience": 5,
+        "threshold": 0.0001,
+        "threshold_mode": "rel",
+        "cooldown": 0,
+        "min_lr": 0,
+        "eps": 1e-08
+      }
+    },
+    "balance_classes": false,
+    "resize": [
+      224,
+      224
+    ],
+    "resize_interpolation": "nearest",
+    "expand": 0,
+    "label_dict": {
+      "Abies": 0,
+      "Acer": 1,
+      "Alnus": 2,
+      "Amelanchier": 3,
+      "Arctostaphylos": 4,
+      "Betula": 5,
+      "Bourreria": 6,
+      "Bucida": 7,
+      "Bursera": 8,
+      "Calocedrus": 9,
+      "Carpinus": 10,
+      "Carya": 11,
+      "Celtis": 12,
+      "Cercis": 13,
+      "Cornus": 14,
+      "Diospyros": 15,
+      "Fagus": 16,
+      "Fraxinus": 17,
+      "Gleditsia": 18,
+      "Gymnocladus": 19,
+      "Halesia": 20,
+      "Ilex": 21,
+      "Juglans": 22,
+      "Juniperus": 23,
+      "Larix": 24,
+      "Liquidambar": 25,
+      "Liriodendron": 26,
+      "Maclura": 27,
+      "Magnolia": 28,
+      "Metrosideros": 29,
+      "Morus": 30,
+      "Nyssa": 31,
+      "Ostrya": 32,
+      "Oxydendrum": 33,
+      "Picea": 34,
+      "Pinaceae": 35,
+      "Pinus": 36,
+      "Pisonia": 37,
+      "Platanus": 38,
+      "Populus": 39,
+      "Prunus": 40,
+      "Pseudotsuga": 41,
+      "Quercus": 42,
+      "Robinia": 43,
+      "Sassafras": 44,
+      "Sideroxylon": 45,
+      "Symphoricarpos": 46,
+      "Taxus": 47,
+      "Thuja": 48,
+      "Tilia": 49,
+      "Triadica": 50,
+      "Tsuga": 51,
+      "Ulmus": 52,
+      "Unknown": 53
+    }
+  }
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5947b83bc5d186b7069439c6dcfbf885cf151a04f25ed766456784f758584c71
+size 44866920