Push model using huggingface_hub.

README.md

@@ -1,23 +1,10 @@
 ---
-license: apache-2.0
 tags:
-- deepforest
+- model_hub_mixin
+- pytorch_model_hub_mixin
 ---
 
-A model to be used in conjunction with the DeepForest python package for airborne machine learning. This is a Alive/Dead tree crown classifier. Tree crowns are first detected seperately, see https://huggingface.co/weecology/deepforest-tree. Then the locations are passed to this model to classify alive/dead status.
-
-For more information see https://deepforest.readthedocs.io/en/latest/prebuilt.html#alive-dead-trees, as well as the general crop classifier docs https://deepforest.readthedocs.io/en/latest/CropModels.html
-
-To provide a simple filter for trees that appear dead in the RGB data we collected 6,342 image crops from the prediction landscape, as well as other NEON sites, and hand annotated them as either alive or dead. We finetuned a resnet-50 pre-trained on ImageNet to classify alive or dead trees before passing them to the species classification model. The model was trained with an ADAM optimizer with a learning rate of 0.001 and batch size of 128 for 40 epochs, and was evaluated on a randomly held out of 10% of the crops. The evaluation accuracy of the alive-dead model was 95.8% (Table S1).
-
-Table S1 Confusion matrix for the Alive/Dead model in Weinstein et al. 2023
-
-        |        Predicted | Alive | Dead |
-        |-----------------|-------|------|
-        | Observed        | 527   | 9    |
-        |                 | 10    | 89   |
-
-* Note *, due to the smaller training sizes, the confidence scores are overfit and not smooth. We do not recommend using the confidence scores from this model until it is trained on more diverse data.
-
-
-Citation: Weinstein, Ben G., et al. "Capturing long‐tailed individual tree diversity using an airborne imaging and a multi‐temporal hierarchical model." Remote Sensing in Ecology and Conservation 9.5 (2023): 656-670.
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Code: [More Information Needed]
+- Paper: [More Information Needed]
+- Docs: [More Information Needed]

config.json

@@ -0,0 +1,8 @@
+{
+  "batch_size": 4,
+  "label_dict": null,
+  "lr": 0.0001,
+  "model": null,
+  "num_classes": 2,
+  "num_workers": 0
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4258183b8b90bbce9dcc69ea46e7aac86d892aac34ee0602fa3dc2eafe75eb21
+size 94291856