[WIP] Integration with DeepLabCut 3.0 - PyTorch Engine#121

Open

n-poulsen wants to merge 66 commits intomainfrom

Commits on Feb 24, 2025

Remove TensorFlow references

dikraMasrour
authored and
n-poulsen
committed
Add comments on TF questions for changing dlc-live pipeline to Pytorch

AnnaStuckert
authored and
n-poulsen
committed
Vanilla pytorch inference done; Commenting out TensorFlow references in code

dikraMasrour
authored and
n-poulsen
committed
change testing directory for Anna and allow to run on CPU

AnnaStuckert
authored and
n-poulsen
committed
Fix display, clean code, adapt frame processing, onnx inference

dikraMasrour
authored and
n-poulsen
committed
add screenshots
n-poulsen
committed
Fix CPU inference crash + GPU (cuda) & CPU support for Pytorch and ONNX inference

dikraMasrour
authored and
n-poulsen
committed
Video analysis feature

AnnaStuckert
authored and
n-poulsen
committed
Improvements on benchmark_pytorch.py

AnnaStuckert
authored and
n-poulsen
committed
Implement TensorRT optimisation on ONNX models and FP16 precision inference

dikraMasrour
authored and
n-poulsen
committed
Avs live feed

AnnaStuckert
authored and
n-poulsen
committed
Tutorial notebook in progress

dikraMasrour
authored and
n-poulsen
committed
bug fixing h5 saving for live video feed

AnnaStuckert
authored and
n-poulsen
committed
add code to save numbers in csv and h5 as numbers, not tensor(number)

AnnaStuckert
authored and
n-poulsen
committed
add timestamp suffix to videos and csv/h5 files

AnnaStuckert
authored and
n-poulsen
committed
fix live inference and display, black and isort

dikraMasrour
authored and
n-poulsen
committed
cleaning out unused files

AnnaStuckert
authored and
n-poulsen
committed
update docstrings, clean dlclive script

dikraMasrour
authored and
n-poulsen
committed
Continued DeepLabCut-Live implementation for DeepLabCut 3.0
n-poulsen
committed

Commits on May 5, 2025

Update README.md

MMathisLab
and
maximpavliv
authored

Commits on Jun 4, 2025

Commits on Jun 5, 2025

Commits on Jun 15, 2025

Delete poetry.lock
MMathisLab
authored

Commits on Aug 22, 2025

SimCCPredictor: add visilibity computation (#131)
maximpavliv
authored