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

Closed

n-poulsen wants to merge 24 commits into

DeepLabCut:masterfrom

n-poulsen:dlclive3

Commits on Feb 24, 2025

Remove TensorFlow references

dikraMasrour
authored and
Niels Poulsen
committed
b11743c
Add comments on TF questions for changing dlc-live pipeline to Pytorch

AnnaStuckert
authored and
Niels Poulsen
committed
c8f5839
Vanilla pytorch inference done; Commenting out TensorFlow references in code

dikraMasrour
authored and
Niels Poulsen
committed
030c42c
change testing directory for Anna and allow to run on CPU

AnnaStuckert
authored and
Niels Poulsen
committed
d75339a
Fix display, clean code, adapt frame processing, onnx inference

dikraMasrour
authored and
Niels Poulsen
committed
4de5829
add screenshots
Niels Poulsen
committed
40a9dd4
Fix CPU inference crash + GPU (cuda) & CPU support for Pytorch and ONNX inference

dikraMasrour
authored and
Niels Poulsen
committed
2fc48f1
Video analysis feature

AnnaStuckert
authored and
Niels Poulsen
committed
125a072
Improvements on benchmark_pytorch.py

AnnaStuckert
authored and
Niels Poulsen
committed
9d2350e
Implement TensorRT optimisation on ONNX models and FP16 precision inference

dikraMasrour
authored and
Niels Poulsen
committed
13b8fae
Avs live feed

AnnaStuckert
authored and
Niels Poulsen
committed
8b41df6
Tutorial notebook in progress

dikraMasrour
authored and
Niels Poulsen
committed
d1e7df9
bug fixing h5 saving for live video feed

AnnaStuckert
authored and
Niels Poulsen
committed
2fb39fc
add code to save numbers in csv and h5 as numbers, not tensor(number)

AnnaStuckert
authored and
Niels Poulsen
committed
98b3a13
add timestamp suffix to videos and csv/h5 files

AnnaStuckert
authored and
Niels Poulsen
committed
e6a914a
fix live inference and display, black and isort

dikraMasrour
authored and
Niels Poulsen
committed
29c8122
cleaning out unused files

AnnaStuckert
authored and
Niels Poulsen
committed
9508a79
update docstrings, clean dlclive script

dikraMasrour
authored and
Niels Poulsen
committed
d61f892
Continued DeepLabCut-Live implementation for DeepLabCut 3.0
Niels Poulsen
committed
f527d38

Commits on Feb 27, 2025

working on README
Niels Poulsen
committed
4c07e04
improved docs
Niels Poulsen
committed
82daf43
improved docs for PyTorch code
Niels Poulsen
committed
40005ae
improved readme
Niels Poulsen
committed
c6a1f69
fix default top down dynamic cropping parameters
Niels Poulsen
committed
d738a2c

Navigation Menu

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

[WIP] Integration with DeepLabCut 3.0 - PyTorch Engine#120
n-poulsen wants to merge 24 commits into
DeepLabCut:masterfrom
n-poulsen:dlclive3

Commits on Feb 24, 2025

Remove TensorFlow references

Add comments on TF questions for changing dlc-live pipeline to Pytorch

Vanilla pytorch inference done; Commenting out TensorFlow references in code

change testing directory for Anna and allow to run on CPU

Fix display, clean code, adapt frame processing, onnx inference

add screenshots

Fix CPU inference crash + GPU (cuda) & CPU support for Pytorch and ONNX inference

Video analysis feature

Improvements on benchmark_pytorch.py

Implement TensorRT optimisation on ONNX models and FP16 precision inference

Avs live feed

Tutorial notebook in progress

bug fixing h5 saving for live video feed

add code to save numbers in csv and h5 as numbers, not tensor(number)

add timestamp suffix to videos and csv/h5 files

fix live inference and display, black and isort

cleaning out unused files

update docstrings, clean dlclive script

Continued DeepLabCut-Live implementation for DeepLabCut 3.0

Commits on Feb 27, 2025

working on README

improved docs

improved docs for PyTorch code

improved readme

fix default top down dynamic cropping parameters