Upload 7 files

README.md

@@ -0,0 +1,24 @@
+---
+license: apache-2.0
+tags:
+- honeybee
+- image
+- insect
+- bumblebee
+- image classification
+- autogluon
+- autogluon 1.1.1
+- transformers
+---
+
+# Model Card for Model ID
+
+This application uses a MultiModalPredictor from the AutoGluon library to classify images of bees.
+The model is fine-tuned on a curated dataset from inaturalist containting over 70,000 images and achieved an accuracy of 97.5%. 
+The model evaluates the image and predicts whether it contains a honey bee, bumblebee, or a vespidae (wasp, hornet, etc.).
+
+You may interact with this model at https://honeybeeclassification.streamlit.app/
+
+
+- **Developed by:** Ian Horvath
+- **Funded by:** The Honey Bee Society

assets.json

@@ -0,0 +1 @@
+{"class_name": "BaseLearner", "column_types": {"image": "image_path", "label": "categorical"}, "label_column": "label", "problem_type": "multiclass", "presets": null, "eval_metric_name": "accuracy", "validation_metric_name": "accuracy", "minmax_mode": "max", "output_shape": 3, "save_path": "./tmp/2f5bfeedd28a488d8f940e6568bea15b-beeml", "pretrained": true, "pretrained_path": null, "fit_called": true, "best_score": 0.9559999704360962, "total_train_time": 18989.389338970184, "version": "1.1.1"}

config.yaml

@@ -0,0 +1,122 @@
+model:
+  names:
+  - timm_image
+  timm_image:
+    checkpoint_name: caformer_b36.sail_in22k_ft_in1k
+    mix_choice: all_logits
+    data_types:
+    - image
+    train_transforms:
+    - resize_shorter_side
+    - center_crop
+    - trivial_augment
+    val_transforms:
+    - resize_shorter_side
+    - center_crop
+    image_norm: imagenet
+    image_size: null
+    max_img_num_per_col: 2
+data:
+  image:
+    missing_value_strategy: zero
+  text:
+    normalize_text: false
+  categorical:
+    minimum_cat_count: 100
+    maximum_num_cat: 20
+    convert_to_text: false
+  numerical:
+    convert_to_text: false
+    scaler_with_mean: true
+    scaler_with_std: true
+  document:
+    missing_value_strategy: zero
+  label:
+    numerical_label_preprocessing: standardscaler
+  pos_label: null
+  column_features_pooling_mode: concat
+  mixup:
+    turn_on: false
+    mixup_alpha: 0.8
+    cutmix_alpha: 1.0
+    cutmix_minmax: null
+    prob: 1.0
+    switch_prob: 0.5
+    mode: batch
+    turn_off_epoch: 5
+    label_smoothing: 0.1
+  templates:
+    turn_on: false
+    num_templates: 30
+    template_length: 2048
+    preset_templates:
+    - super_glue
+    - rte
+    custom_templates: null
+optimization:
+  optim_type: adamw
+  learning_rate: 0.0001
+  weight_decay: 0.001
+  lr_choice: layerwise_decay
+  lr_decay: 0.9
+  lr_schedule: cosine_decay
+  max_epochs: 20
+  max_steps: -1
+  warmup_steps: 0.1
+  end_lr: 0
+  lr_mult: 1
+  patience: 10
+  val_check_interval: 0.5
+  check_val_every_n_epoch: 1
+  skip_final_val: false
+  gradient_clip_val: 1
+  gradient_clip_algorithm: norm
+  track_grad_norm: -1
+  log_every_n_steps: 10
+  top_k: 3
+  top_k_average_method: greedy_soup
+  efficient_finetune: null
+  lora:
+    module_filter: null
+    filter:
+    - query
+    - value
+    - ^q$
+    - ^v$
+    - ^k$
+    - ^o$
+    r: 8
+    alpha: 8
+    conv_lora_expert_num: 8
+  loss_function: auto
+  focal_loss:
+    alpha: null
+    gamma: 2.0
+    reduction: mean
+  mask2former_loss:
+    loss_cross_entropy_weight: 10.0
+    loss_mask_weight: 5.0
+    loss_dice_weight: 5.0
+  extra_trainable_params: []
+env:
+  num_gpus: 0
+  num_nodes: 1
+  batch_size: 128
+  per_gpu_batch_size: 8
+  eval_batch_size_ratio: 4
+  per_gpu_batch_size_evaluation: null
+  precision: 32
+  num_workers: 2
+  num_workers_evaluation: 2
+  accelerator: auto
+  fast_dev_run: false
+  deterministic: false
+  auto_select_gpus: true
+  strategy: auto
+  deepspeed_allgather_size: 1000000000.0
+  deepspeed_allreduce_size: 1000000000.0
+  compile:
+    turn_on: false
+    mode: default
+    dynamic: true
+    backend: inductor

df_preprocessor.pkl

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

gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

hparams.yaml

@@ -0,0 +1,15 @@
+optim_type: adamw
+lr_choice: layerwise_decay
+lr_schedule: cosine_decay
+lr: 0.0001
+lr_decay: 0.9
+end_lr: 0
+lr_mult: 1
+weight_decay: 0.001
+warmup_steps: 0.1
+validation_metric_name: accuracy
+custom_metric_func: null
+efficient_finetune: null
+mixup_off_epoch: 5
+skip_final_val: false
+track_grad_norm: -1

model.ckpt

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