[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
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

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

[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