detr-finetuned-thermal-dogs-and-people

faldeus0092

Detección de objetos

Este modelo es una versión ajustada de DETR en el conjunto de datos Roboflow Thermal Dogs and People. Logra los siguientes resultados en el conjunto de evaluación: Precisión Promedio (AP) @[ IoU=0.50:0.95 | área= todo | maxDets=100 ] = 0.681, Precisión Promedio (AP) @[ IoU=0.50 | área= todo | maxDets=100 ] = 0.870, Precisión Promedio (AP) @[ IoU=0.75 | área = todo | maxDets=100 ] = 0.778, Precisión Promedio (AP) @[ IoU=0.50:0.95 | área= pequeña | maxDets=100 ] = 0.189, Precisión Promedio (AP) @[ IoU=0.50:0.95 | área= mediana | maxDets=100 ] = 0.489, Precisión Promedio (AP) @[ IoU=0.50:0.95 | área= grande | maxDets=100 ] = 0.720, Rappel Promedio (AR) @[ IoU=0.50:0.95 | área= todo | maxDets= 1 ] = 0.641, Rappel Promedio (AR) @[ IoU=0.50:0.95 | área= todo | maxDets= 10 ] = 0.733, Rappel Promedio (AR) @[ IoU=0.50:0.95 | área= todo | maxDets=100 ] = 0.746, Rappel Promedio (AR) @[ IoU=0.50:0.95 | área= pequeña | maxDets=100 ] = 0.500, Rappel Promedio (AR) @[ IoU=0.50:0.95 | área= mediana | maxDets=100 ] = 0.542, Rappel Promedio (AR) @[ IoU=0.50:0.95 | área= grande | maxDets=100 ] = 0.794.

Como usar

# Usar un pipeline como una ayuda de alto nivel
from transformers import pipeline

image = Image.open('/content/Thermal-Dogs-and-People-1/test/IMG_0006 5_jpg.rf.cd46e6a862d6ffb7fce6795067ce7cc7.jpg')
# image = Image.open(requests.get(url, stream=True).raw) # si deseas abrir desde la URL

obj_detector = pipeline("object-detection", model="faldeus0092/detr-finetuned-thermal-dogs-and-people")

draw = ImageDraw.Draw(image)

for score, label, box in zip(results["scores"], results["labels"], results["boxes"]):
    box = [round(i, 2) for i in box.tolist()]
    x, y, x2, y2 = tuple(box)
    draw.rectangle((x, y, x2, y2), outline="red", width=1)
    draw.text((x, y), model.config.id2label[label.item()], fill="white")

image

Funcionalidades

Detección de objetos
Transformadores
PyTorch
detr
Puntos de Inferencia

Casos de uso

Aprendizaje
Monitoreo del rendimiento de máquinas
Visión en condiciones de poca luz
Imágenes infrarrojas en seguridad
Detección de vida silvestre
Recreación al aire libre