Skip to content

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

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 24 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
24 commits
Select commit Hold shift + click to select a range
b11743c
Remove TensorFlow references
dikraMasrour Aug 9, 2024
c8f5839
Add comments on TF questions for changing dlc-live pipeline to Pytorch
AnnaStuckert Feb 24, 2025
030c42c
Vanilla pytorch inference done; Commenting out TensorFlow references …
dikraMasrour Feb 24, 2025
d75339a
change testing directory for Anna and allow to run on CPU
AnnaStuckert Feb 24, 2025
4de5829
Fix display, clean code, adapt frame processing, onnx inference
dikraMasrour Feb 24, 2025
40a9dd4
add screenshots
n-poulsen Feb 24, 2025
2fc48f1
Fix CPU inference crash + GPU (cuda) & CPU support for Pytorch and ON…
dikraMasrour Feb 24, 2025
125a072
Video analysis feature
AnnaStuckert Feb 24, 2025
9d2350e
Improvements on benchmark_pytorch.py
AnnaStuckert Feb 24, 2025
13b8fae
Implement TensorRT optimisation on ONNX models and FP16 precision inf…
dikraMasrour Feb 24, 2025
8b41df6
Avs live feed
AnnaStuckert Feb 24, 2025
d1e7df9
Tutorial notebook in progress
dikraMasrour Feb 24, 2025
2fb39fc
bug fixing h5 saving for live video feed
AnnaStuckert Feb 24, 2025
98b3a13
add code to save numbers in csv and h5 as numbers, not tensor(number)
AnnaStuckert Feb 24, 2025
e6a914a
add timestamp suffix to videos and csv/h5 files
AnnaStuckert Feb 24, 2025
29c8122
fix live inference and display, black and isort
dikraMasrour Feb 24, 2025
9508a79
cleaning out unused files
AnnaStuckert Feb 24, 2025
d61f892
update docstrings, clean dlclive script
dikraMasrour Feb 24, 2025
f527d38
Continued DeepLabCut-Live implementation for DeepLabCut 3.0
n-poulsen Sep 18, 2024
4c07e04
working on README
n-poulsen Feb 27, 2025
82daf43
improved docs
n-poulsen Feb 27, 2025
40005ae
improved docs for PyTorch code
n-poulsen Feb 27, 2025
c6a1f69
improved readme
n-poulsen Feb 27, 2025
d738a2c
fix default top down dynamic cropping parameters
n-poulsen Feb 27, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
261 changes: 193 additions & 68 deletions README.md

Large diffs are not rendered by default.

6 changes: 3 additions & 3 deletions dlclive/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
Licensed under GNU Lesser General Public License v3.0
"""

from dlclive.version import __version__, VERSION
from dlclive.display import Display
from dlclive.dlclive import DLCLive
from dlclive.processor import Processor
from dlclive.benchmark import benchmark, benchmark_videos, download_benchmarking_data
from dlclive.processor.processor import Processor
from dlclive.version import VERSION, __version__
Loading
Loading