Commit Graph

11 Commits (4ceb6e8a4b86605cdeb6cd7087d1d7ef34452d85)

Author SHA1 Message Date
青龍聖者@bdsqlsz 79948d8c29
Add Animal Pose Preprocessor (RTMPose AP-10K) (#2293)
* add support for animal pose

* Update __init__.py

* fix typo

* flatten json result list

* format files

* remove unused functions

* 🚧 Support edit

* nits

* fix test expecetations

---------

Co-authored-by: huchenlei <chenlei.hu@mail.utoronto.ca>
2023-12-23 15:55:21 -05:00
Chenlei Hu 0cfc88b6a8
🐛 DWPose face keypoint num fix (#1917) 2023-08-13 20:07:03 -04:00
Chenlei Hu d67f31a3a7
🐛 Handle no detection case (#1904) 2023-08-10 21:56:41 -04:00
Chenlei Hu 4fa9190436
Reland Reland "Port DW Pose preprocessor" (#1892)
* Port DW Pose preprocessor (#1856)

*  Add dependencies

* 🚧 wip

* 🚧 wip

* 🚧 download models

* 🚧 Minor fixes

* 🔧 update gitignore

* 🐛 Fix normalization issue

* 🚧 load DW model only when DW preprocessor is selected

*  Change test config

* 🎨 nits

* 🐛 Fix A1111 safe torch issue

📝 v1.1.235 (#1859)

Revert "Port DW Pose preprocessor (#1856)" (#1860)

This reverts commit 0d3310f599.

Reland "Port DW Pose preprocessor" (#1861)

* Revert "Revert "Port DW Pose preprocessor (#1856)" (#1860)"

This reverts commit 17e100e1fe.

* 🐛 Fix install.py

📝 v1.1.236 (#1862)

🐛 Delay import of mmpose (#1866)

📝 v1.1.237 (#1868)

🐛 Fix all keypoints invalid issue (#1871)

🐛 lazy import

🚧 update test expectation

🚧 Switch to onnx

🚧 solve onnx package issue

🔧 Check cuda in more efficient way

🎨 Format code

🔧 Make onnx runtime optional

* Use cv2 to load and run model on cpu

* nit
2023-08-09 00:21:18 -04:00
Chenlei Hu 4fd548b4d2
Revert "Reland dw pose onnx (#1890)" (#1891)
This reverts commit e141d33be9.
2023-08-07 17:50:12 -04:00
Chenlei Hu e141d33be9
Reland dw pose onnx (#1890)
* Port DW Pose preprocessor (#1856)

*  Add dependencies

* 🚧 wip

* 🚧 wip

* 🚧 download models

* 🚧 Minor fixes

* 🔧 update gitignore

* 🐛 Fix normalization issue

* 🚧 load DW model only when DW preprocessor is selected

*  Change test config

* 🎨 nits

* 🐛 Fix A1111 safe torch issue

📝 v1.1.235 (#1859)

Revert "Port DW Pose preprocessor (#1856)" (#1860)

This reverts commit 0d3310f599.

Reland "Port DW Pose preprocessor" (#1861)

* Revert "Revert "Port DW Pose preprocessor (#1856)" (#1860)"

This reverts commit 17e100e1fe.

* 🐛 Fix install.py

📝 v1.1.236 (#1862)

🐛 Delay import of mmpose (#1866)

📝 v1.1.237 (#1868)

🐛 Fix all keypoints invalid issue (#1871)

🐛 lazy import

* 🚧 update test expectation

* 🚧 Switch to onnx

* 🚧 solve onnx package issue

* 🔧 Check cuda in more efficient way

* 🎨 Format code
2023-08-07 17:35:37 -04:00
lvmin 20d25bb546 Revert "Port DW Pose preprocessor (#1856)"
This reverts commit 0d3310f599.
2023-08-05 02:58:03 -07:00
Chenlei Hu 82a6213631
🐛 Fix all keypoints invalid issue (#1871) 2023-08-04 21:41:50 -04:00
Chenlei Hu 8f44b22ee5
Reland "Port DW Pose preprocessor" (#1861)
* Revert "Revert "Port DW Pose preprocessor (#1856)" (#1860)"

This reverts commit 17e100e1fe.

* 🐛 Fix install.py
2023-08-04 16:04:02 -04:00
Chenlei Hu 17e100e1fe
Revert "Port DW Pose preprocessor (#1856)" (#1860)
This reverts commit 0d3310f599.
2023-08-04 15:13:51 -04:00
Chenlei Hu 0d3310f599
Port DW Pose preprocessor (#1856)
*  Add dependencies

* 🚧 wip

* 🚧 wip

* 🚧 download models

* 🚧 Minor fixes

* 🔧 update gitignore

* 🐛 Fix normalization issue

* 🚧 load DW model only when DW preprocessor is selected

*  Change test config

* 🎨 nits

* 🐛 Fix A1111 safe torch issue
2023-08-04 15:02:58 -04:00