Commit Graph

20 Commits (442398bb9fdc8f42e48ce02636242933a855afee)

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 9e61022428
🔧 Support SDXL in tests (#2254)
* 🔧 Support SDXL in tests

* 📝 docs

* nits
2023-11-12 14:12:54 -05:00
Chenlei Hu 2d73f31f1b
🐛 Disable flaky tests (#2232) 2023-11-04 23:17:09 -04:00
Chenlei Hu 27c522029e
Fix CI tests (#2220)
* 🔧 setup CI

* 🐛 Fix path

* cd

* set working dir

* fix openpose tests

* Fix web_api tests

* Fix batch hijack tests

* change coverage dir

* 📝 Add doc
2023-11-04 00:48:22 -04:00
Chenlei Hu 664ac74cae
Relands #2033 (#2038)
* 🎨 re-format api.py

* Add pose json to api response

* Simplify response format

* fix
2023-09-04 09:06:46 -04:00
lllyasviel 7bdcb9079a Revert "Add openpose json to API response (#2033)"
This reverts commit e2cd3b91a8.
2023-09-04 05:38:19 -07:00
Chenlei Hu e2cd3b91a8
Add openpose json to API response (#2033)
* 🎨 re-format api.py

* Add pose json to api response

* Simplify response format
2023-09-03 17:33:05 -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
Chenlei Hu af34f51449
🐛 Fix cv2 color channel issue in unittest (#1878) 2023-08-05 20:16:23 -04:00
lvmin 20d25bb546 Revert "Port DW Pose preprocessor (#1856)"
This reverts commit 0d3310f599.
2023-08-05 02:58:03 -07: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
huchenlei e6b6993b6d 🐛 handles np float types when serializing JSON 2023-04-28 22:05:09 -04:00
huchenlei 3875a16abe Add json encoding logic 2023-04-28 17:35:50 -04:00
huchenlei 09296b1684 Refactor openpose data representation 2023-04-28 16:30:04 -04:00
huchenlei 55171dc9f9 Add end to end unittest on openpose preprocessor 2023-04-28 14:57:55 -04:00
huchenlei 99cda2f13e Rename dir to avoid module name conflict 2023-04-27 23:42:25 -04:00