protege-med / README.md
langutang's picture
Update README.md
8390f82 verified
---
license: mit
language:
- en
metrics:
- accuracy
pipeline_tag: robotics
tags:
- robotics
- opencv
- tensorflow
---
# 🧠 Protoge-Med: Vision-Based Detection & Tracking with TensorFlow
[![TensorFlow](https://img.shields.io/badge/framework-TensorFlow-orange)](https://www.tensorflow.org/)
[![Computer Vision](https://img.shields.io/badge/task-Object%20Detection%20%26%20Tracking-blue)]()
[![Classes](https://img.shields.io/badge/labels-1000%20Supported-green)]()
**Protoge-Med** is a powerful object detection and tracking model built on TensorFlow, designed for large-scale multi-class visual recognition and tracking in real-time systems. It supports simultaneous detection of **all 1000+ labels** or can be configured to track a **custom subset of interest**.
---
## πŸš€ Key Features
- 🧠 Detects and tracks up to **1000 unique object categories**
- 🎯 Supports both **full-scope detection** and **targeted label tracking**
- βš™οΈ Real-time tracking with TensorFlow and OpenCV integration
- πŸ’‘ Modular design for flexible use in healthcare, robotics, surveillance, and more
- πŸ“¦ Exportable to TensorFlow Lite and TF.js for cross-platform deployment
---
## πŸ₯ Intended Use Cases
- Medical robotics and assistive devices
- Smart hospital environments
- Video surveillance and tracking
- Multi-object tracking in clinical or industrial settings
---
## πŸ“¦ How to Use
```python
import tensorflow as tf
import cv2
# Load the model
model = tf.saved_model.load("path/to/protoge-med")
# Provide a list of custom labels (optional)
target_labels = ["stethoscope", "syringe", "mask"]
# Perform detection and tracking
detections = model(input_tensor, labels=target_labels)
```
---
## πŸ§ͺ Supported Modes
1. **Full Detection Mode** – Identify and track all 1000 labels in a single frame.
2. **Selective Mode** – Focus on a specified list of labels to reduce computational load and improve accuracy.
---
## πŸ“Š Performance Metrics
| Metric | Value |
|-------------------|---------------|
| Classes Supported | 1000+ |
| Tracking Speed | ~30 FPS |
| Inference Time | < 60ms/frame |
| Model Size | ~40MB |
---
## 🧠 Training Details
Protoge-Med was trained on a hybrid dataset combining:
- Extended COCO and Open Images datasets
- Domain-specific annotations for medical tools and equipment
- Augmented for tracking stability and occlusion handling
---
## πŸ“– Citation
If you use **Protoge-Med** in your research or applications, please cite:
```
@misc{protogemed2025,
title={Protoge-Med: Scalable Real-Time Detection and Tracking with TensorFlow},
author={Lang, John},
year={2025},
howpublished={\url{https://huggingface.co/langutang/protoge-med}}
}
```
---
## πŸ“¬ Contact & License
- πŸ“« Reach out for support or contributions via Hugging Face issues.
- βš–οΈ License: Apache 2.0
---
## πŸ€– Hugging Face Integration
```python
from transformers import AutoFeatureExtractor, TFModelForObjectDetection
model = TFModelForObjectDetection.from_pretrained("langutang/protoge-med")
extractor = AutoFeatureExtractor.from_pretrained("langutang/protoge-med")
```
---
Make large-scale visual tracking intelligent with **Protoge-Med** πŸ§