Upload finetuned_checkpoint.ckpt

#4
by camappel - opened

This PR includes the updated .ckpt file for the livestock detection model, with significant performance improvements after fine-tuning for 5 epochs. The model was trained on labelled aerial images of cows from Dataverse (2585 training/808 validation), and its performance metrics on the test set (subset of images excluded from training/validation sets) showed substantial gains:

Box Recall: Improved from 0.4405 to 0.9535.
Box Precision: Improved from 0.5826 to 0.8587.
Mean IoU: Improved from 0.3135 to 0.6571.

weecology org

Thanks for this work @camappel , I'd like to make sure you get the attribution. I am going to download the checkpoint, upload it as .safetensors format that hugging face wants for our system, and then close this PR. I will add your attribution to the DeepForest docs.

bw4sz changed pull request status to merged

Sign up or log in to comment