Add method to export PyTorch DeepLabCut model to ONNX by Yu-changqing · Pull Request #3147 · DeepLabCut/DeepLabCut

@Yu-changqing

This script provides a straightforward and reliable method to export a DeepLabCut PyTorch pose estimation model to ONNX.

@Yu-changqing

@AlexEMG

Excellent! Do you also have code for running inference?

@Yu-changqing

Do you also have code for running inference?

Yeah, I’ve been writing some inference code for Dlc,but it’s still a bit rough at the moment.

@MMathisLab

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Copilot encountered an error and was unable to review this pull request. You can try again by re-requesting a review.

@deruyter92