๐Ÿญ DeepRodent

A Robust and Generalizable Vision Framework for Automated Rodent Monitoring in Experimental Biology

Teerapong Panboonyuen (@kaopanboonyuen)

Department of Computer Science, Khon Kaen University ยท PBYAIL ยท Chulalongkorn University

๐Ÿ“ง teerapong.panboonyuen@gmail.com

GitHub Project Page Demo License


โœจ Overview

DeepRodent is a unified, detector-agnostic deep learning framework built for precise, generalizable, multi-task rodent monitoring in laboratory environments. It jointly predicts:

  • ๐Ÿ“ฆ Standard object detection
  • ๐Ÿ”„ Oriented Bounding Boxes (OBB) โ€” captures rotation during grooming, rearing, and huddling
  • ๐ŸŽจ Pixel-level instance segmentation โ€” accurate boundary masks even under occlusion
  • ๐Ÿงญ Temporal-consistent behavioral embeddings โ€” stable identity tracking across video frames

Rather than bolting these on separately, DeepRodent fuses them into a single-stage, multi-head architecture โ€” so one forward pass gives you everything downstream biology pipelines need.

๐ŸŽฏ Metric ๐Ÿฅ‡ DeepRodent Best Baseline (YOLO12-Seg) ๐Ÿ“ˆ Gain
mAPโ‚…โ‚€ 96.2 94.4 +1.8
mAPโ‚…โ‚€โ‚‹โ‚‰โ‚… 84.6 78.2 +6.4
Cross-domain (Lab C) 91.5 87.2 +4.3
FPS (real-time) 154 156 โ€”

๐Ÿš€ Quick Start

pip install ultralytics
from ultralytics import YOLO

# ๐Ÿญ Load DeepRodent
model = YOLO("DeepRodent_WEIGHT.pt")

# ๐Ÿ” Run inference
results = model.predict("your_cage_image.jpg", save=True)

# ๐ŸŽจ Visualize instance masks + OBB
results[0].show()

๐Ÿ“ฅ Download directly

from huggingface_hub import hf_hub_download

weight_path = hf_hub_download(
    repo_id="kaopanboonyuen/DeepRodent",
    filename="DeepRodent_WEIGHT.pt"
)

๐Ÿง  Why DeepRodent?

Challenge in the Lab ๐Ÿ”ฌ How DeepRodent Solves It ๐Ÿ’ก
Rodents curl, rear, and huddle Oriented Bounding Boxes capture true rotation
Axis-aligned boxes overlap during interaction Instance segmentation isolates each animal precisely
Lighting/cage geometry varies across labs Cross-domain robustness constraint for generalization
Jittery masks across video frames Temporal consistency loss stabilizes identity tracking
Need for downstream analytics Direct output โ†’ trajectories, heatmaps, behavior states

๐Ÿ—๏ธ Architecture Snapshot

Input Frame โ†’ Shared Backbone โ†’ โ”ฌโ”€โ”€ Detection Head
                                 โ”œโ”€โ”€ OBB Head (ฮธ-aware)
                                 โ”œโ”€โ”€ Instance Seg Head
                                 โ””โ”€โ”€ Temporal Embedding Head
                                          โ†“
                        Trajectory Tracking ยท Behavior States ยท Occupancy Heatmaps

Detector-agnostic: plug into YOLOv8 โ†’ YOLO12 families and gain +2.6 to +3.1 mAP consistently. ๐Ÿ”Œ


๐Ÿ“Š Benchmarks

Evaluated on a private multi-setting laboratory rodent dataset (~30K annotated frames, 60/20/20 split):

  • ๐Ÿฅ‡ Outperforms RT-DETR by +6.7 mAPโ‚…โ‚€โ‚‹โ‚‰โ‚…
  • ๐Ÿฅ‡ Outperforms Mask R-CNN by +13.5 mAPโ‚…โ‚€โ‚‹โ‚‰โ‚…
  • โšก Maintains 154 FPS โ€” real-time, deployment-ready

Full ablation, cross-domain, and SOTA comparison tables are in the paper.


๐Ÿ“ Repository Links

Resource Link
๐Ÿ’ป Code github.com/kaopanboonyuen/DeepRodent
๐ŸŒ Project Page kaopanboonyuen.github.io/DeepRodent
๐Ÿค— Weights (this repo) DeepRodent_WEIGHT.pt
๐ŸŽฎ Interactive Demo Hugging Face Spaces

๐Ÿพ Ethical Use

DeepRodent is an assistive research tool โ€” it does not replace veterinary oversight or trained behavioral experts. Developed following ARRIVE guidelines and the 3Rs principles (Replacement, Reduction, Refinement). Please validate performance before deployment in a new laboratory setting.


๐Ÿ“œ Citation

If DeepRodent helps your research, please cite:

@article{panboonyuen2026deeprodent,
  title   = {DeepRodent: A Robust and Generalizable Vision Framework for Automated Rodent Monitoring in Experimental Biology},
  author  = {Panboonyuen, Teerapong},
  year    = {2026}
}

Made with ๐Ÿงก for open, reproducible science.

โญ If this helped your research, consider starring the GitHub repo!

Downloads last month
15
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Space using kaopanboonyuen/DeepRodent 1