Downloads · 30 days
10
42% of all-time downloads
Monuaub/distilbert-finetuned-visdrone-text
distilbert-finetuned-visdrone-text is a machine learning model from Monuaub. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This model is a DistilBERT fine-tuned model for text classification of drone surveillance messages, specifically for scenarios inspired by the VisDrone dataset.
Downloads · 30 days
10
42% of all-time downloads
All-time downloads
24
Public
Parameters
67M
268 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors268 MB · 100%
From the Hugging Face model README
This model is a DistilBERT fine-tuned model for text classification of drone surveillance messages, specifically for scenarios inspired by the VisDrone dataset.
Use this model to classify drone-generated descriptions into meaningful categories, such as:
from transformers import pipeline
model = "Monuaub/distilbert-finetuned-visdrone-text"
clf = pipeline("text-classification", model=model, tokenizer=model)
text = "The drone successfully tracked the vehicle across the terrain."
result = clf(text)
print(result)