Commit Graph

1135 Commits

Author SHA1 Message Date
Prianka Liz Kariat
47013d289e Added flow limiter calculator in MediaPipeTasksCommon 2023-05-04 23:19:12 +05:30
Prianka Liz Kariat
253662149e Updated pixel format types in object detector 2023-05-04 23:03:03 +05:30
Sebastian Schmidt
767db32d69 Support multiple poses for PoseLandmarker
PiperOrigin-RevId: 529430797
2023-05-04 09:29:12 -07:00
Prianka Liz Kariat
3df4f7db64 Updated time out for object detector 2023-05-04 20:00:29 +05:30
Prianka Liz Kariat
08282d9fd7 Updated time out for image classifier async tests 2023-05-04 19:59:39 +05:30
Prianka Liz Kariat
a4e11eac78 Added constants for time out 2023-05-04 19:55:19 +05:30
Prianka Liz Kariat
00712d727e Updated wait time for object detector tests 2023-05-04 19:53:06 +05:30
Prianka Liz Kariat
33ae23c53a Increased wait time for image classifier asynchronous tests 2023-05-04 19:51:54 +05:30
Prianka Liz Kariat
8ec0724b65 Updated documentation to include note about rgba images 2023-05-04 19:44:11 +05:30
Prianka Liz Kariat
ddd1515f88 Updated documentation 2023-05-04 19:40:15 +05:30
Prianka Liz Kariat
d401439daa Updated formatting 2023-05-04 19:22:11 +05:30
Prianka Liz Kariat
1136d4d515 Updated CVPixelBuffer to support pixel format type of 32RGBA 2023-05-04 18:58:49 +05:30
Prianka Liz Kariat
7a7f27c34b Merge branch 'ios-normalized-keypoint-hash' into ios-object-detector-async-fixes 2023-05-04 17:21:36 +05:30
Prianka Liz Kariat
381ffcb474 Added hash implementation for iOS normalized keypoint 2023-05-04 17:10:07 +05:30
Prianka Liz Kariat
87593a2ade Updated docuemntation of MPPObjectDetector 2023-05-04 17:05:17 +05:30
Prianka Liz Kariat
ab135190e5 Updated iOS object detector to use delegates instead of callbacks for async calls 2023-05-04 17:03:40 +05:30
Prianka Liz Kariat
a21c08bf4d Added method for creating unique dispatch queue names in MPPVisionTaskRunner 2023-05-04 17:00:12 +05:30
Prianka Liz Kariat
e47bb16544 Added validation of C++ image classification result packet in MPPImageClassifierResult+Helpers.mm 2023-05-04 16:52:58 +05:30
Prianka Liz Kariat
ab4b07646c Updated MPPImageClassifier to use delegates instead of completion blocks for callback. 2023-05-04 16:43:18 +05:30
Prianka Liz Kariat
1323a5271c Added method to create unique dispatch queue names in MPPVisionTaskRunner 2023-05-04 16:39:43 +05:30
Jiuqiang Tang
c6e3f08282 Expose FaceAligner and LanguageDetector to be public MediaPipe Tasks Python API.
PiperOrigin-RevId: 529227382
2023-05-03 16:38:49 -07:00
Copybara-Service
8c324fbd77 Merge pull request #4325 from kinaryml:language-detector-python
PiperOrigin-RevId: 529213966
2023-05-03 15:50:42 -07:00
MediaPipe Team
a09e39d431 Add TransformerParameters proto
PiperOrigin-RevId: 529213840
2023-05-03 15:46:50 -07:00
Yuqi Li
e428bdb7e8 internal change.
PiperOrigin-RevId: 529181374
2023-05-03 13:37:16 -07:00
MediaPipe Team
606b83ac65 Internal change
PiperOrigin-RevId: 529180655
2023-05-03 13:32:44 -07:00
MediaPipe Team
7c955246aa Support scribble input for Interactive Segmenter Java API
PiperOrigin-RevId: 529177660
2023-05-03 13:22:09 -07:00
Copybara-Service
e84e90e5b2 Merge pull request #4361 from kinaryml:face-aligner-python
PiperOrigin-RevId: 529165597
2023-05-03 12:39:23 -07:00
MediaPipe Team
c780559214 Internal change
PiperOrigin-RevId: 529161249
2023-05-03 12:20:44 -07:00
MediaPipe Team
09662749ea Support scribble input for Interactive Segmenter
PiperOrigin-RevId: 529156049
2023-05-03 12:01:10 -07:00
MediaPipe Team
1dea01aecc Internal change
PiperOrigin-RevId: 528996603
2023-05-02 22:57:38 -07:00
MediaPipe Team
c698381e48 Internal change
PiperOrigin-RevId: 528939095
2023-05-02 18:05:31 -07:00
Jiuqiang Tang
bf11fb313e Expose PoseLandmarker as a public MediaPipe Tasks Python API.
PiperOrigin-RevId: 528882303
2023-05-02 14:04:01 -07:00
Jiuqiang Tang
4d9812af43 Pose detector uses advanced_gpu_api for gpu inference to resolve unsupported gpu op issue.
PiperOrigin-RevId: 528879218
2023-05-02 13:52:26 -07:00
Yuqi Li
9ce16fddeb nit: format the documentation of LandmarksDetectionResult.
PiperOrigin-RevId: 528848566
2023-05-02 11:56:47 -07:00
MediaPipe Team
421c9e8e97 Fix typo
PiperOrigin-RevId: 528829423
2023-05-02 10:51:47 -07:00
Jiuqiang Tang
60055f6fee Add more comments and usage example of the face stylizer graph.
PiperOrigin-RevId: 528823127
2023-05-02 10:32:43 -07:00
Chuo-Ling Chang
3719aaef7e Fix typo.
PiperOrigin-RevId: 528693117
2023-05-01 23:52:45 -07:00
Jiuqiang Tang
fca728d226 Set face alignment image width and hight to 256.
PiperOrigin-RevId: 528583074
2023-05-01 15:01:39 -07:00
Jiuqiang Tang
162a999887 Check the output stream tag rather than the input stream tag in face stylizer graph.
PiperOrigin-RevId: 528555024
2023-05-01 13:14:32 -07:00
Yuqi Li
085f8265fb Internal change
PiperOrigin-RevId: 528517562
2023-05-01 11:01:23 -07:00
Kinar R
544e4b66f7
Merge branch 'google:master' into face-aligner-python 2023-05-01 22:56:00 +05:30
Jiuqiang Tang
b9a9da5de5 Ignore fetching face stylizer model when the graph doesn't output stylized face images.
PiperOrigin-RevId: 528504312
2023-05-01 10:18:41 -07:00
kinaryml
bd039f8b65 Updated necessary BUILD files 2023-05-01 05:56:52 -07:00
kinaryml
209d78f36c Added the Face Aligner Python API 2023-05-01 05:55:46 -07:00
Jiuqiang Tang
c29e43dda0 Add the "FACE_ALIGNMENT" output stream to the face stylizer graph.
PiperOrigin-RevId: 528345204
2023-04-30 16:59:36 -07:00
Sebastian Schmidt
8e510a3255 Invoke PoseListener callback while C++ Packet is still active
PiperOrigin-RevId: 528061429
2023-04-28 21:22:02 -07:00
Sebastian Schmidt
253f13ad62 Invoke callback for InteractiveSegmenter while C++ Packets are active
PiperOrigin-RevId: 528053621
2023-04-28 20:34:17 -07:00
Sebastian Schmidt
d5c5457d25 Only log warnings once if color conversion is not specified
PiperOrigin-RevId: 528052009
2023-04-28 20:24:17 -07:00
Sebastian Schmidt
a9721ae2fb Invoke callback for ImageSegmenter while C++ Packets are active
PiperOrigin-RevId: 528047220
2023-04-28 20:01:10 -07:00
Sebastian Schmidt
e15add2475 Shorten MPImage API
PiperOrigin-RevId: 528039371
2023-04-28 19:00:17 -07:00
Sebastian Schmidt
874cc9dea3 Update PoseLandmarker to return MPImage
PiperOrigin-RevId: 528022223
2023-04-28 17:13:07 -07:00
Sebastian Schmidt
dcef6df1cb Update InteractiveSegmenter to return MPImage
PiperOrigin-RevId: 528010944
2023-04-28 16:13:51 -07:00
Copybara-Service
bbbc0f98c5 Merge pull request #4268 from priankakariatyml:object-detector-objc-tests
PiperOrigin-RevId: 527991967
2023-04-28 14:52:20 -07:00
Sebastian Schmidt
2c1d9c6582 Update ImageSegmenter to return MPImage
PiperOrigin-RevId: 527990991
2023-04-28 14:48:30 -07:00
Sebastian Schmidt
a544098100 Update FaceStylizer to return MPImage
PiperOrigin-RevId: 527980696
2023-04-28 14:05:11 -07:00
Copybara-Service
5cffb3973f Merge pull request #4303 from kinaryml:pose-landmarker-python
PiperOrigin-RevId: 527948047
2023-04-28 11:59:28 -07:00
Copybara-Service
2bb1b454ea Merge pull request #4300 from priankakariatyml:ios-text-cocoapods-force-load
PiperOrigin-RevId: 527932547
2023-04-28 11:06:44 -07:00
MediaPipe Team
b2fbb2ddab Internal change
PiperOrigin-RevId: 527909361
2023-04-28 09:43:56 -07:00
kinaryml
3b06772d9a Fixed BUILD 2023-04-27 21:13:31 -07:00
kinaryml
305866ccae Updated BUILD files to use the open sourced Language Detector model 2023-04-27 21:11:53 -07:00
Kinar R
76c8251faf
Merge branch 'google:master' into language-detector-python 2023-04-28 09:28:54 +05:30
Sebastian Schmidt
5d9761cbfd Update tests and demos to call "close".
PiperOrigin-RevId: 527746909
2023-04-27 18:58:57 -07:00
Sebastian Schmidt
28b9b8d8a3 Open-sources LanguageDetector model.
PiperOrigin-RevId: 527745108
2023-04-27 18:47:07 -07:00
Sebastian Schmidt
5e41d47f3a Add "close()" method to MP Web Tasks
PiperOrigin-RevId: 527726737
2023-04-27 17:16:40 -07:00
Prianka Liz Kariat
bdede4f94e Updated select conditions 2023-04-28 03:26:29 +05:30
MediaPipe Team
3ca2427cc8 Blendshapes graph take smoothed face landmarks as input.
PiperOrigin-RevId: 527640341
2023-04-27 11:46:40 -07:00
Sebastian Schmidt
1b82821f15 Add support for single-channel images to MPImage
PiperOrigin-RevId: 527629970
2023-04-27 11:12:34 -07:00
Sebastian Schmidt
b457060c3a Generify tests for MPImage
PiperOrigin-RevId: 527611864
2023-04-27 10:12:30 -07:00
Sebastian Schmidt
bc3434108e Update MPImage to use containers
PiperOrigin-RevId: 527596164
2023-04-27 09:18:05 -07:00
Jiuqiang Tang
7c70c62465 Fix typo and improve comments.
PiperOrigin-RevId: 527580369
2023-04-27 08:10:29 -07:00
MediaPipe Team
2122b5d7be Internal change
PiperOrigin-RevId: 527430483
2023-04-26 18:17:56 -07:00
MediaPipe Team
a45d1f5e90 Internal change.
PiperOrigin-RevId: 527374728
2023-04-26 14:27:58 -07:00
Prianka Liz Kariat
261e02e491 Fixed case name 2023-04-27 02:55:58 +05:30
Prianka Liz Kariat
ee2665ad13 Added missing input files in vision library 2023-04-27 02:54:50 +05:30
Prianka Liz Kariat
1d8e24b9aa Updated documentation 2023-04-27 02:47:52 +05:30
Prianka Liz Kariat
a8cb1f1dad Updated default values 2023-04-27 02:47:19 +05:30
Prianka Liz Kariat
1e776e8e01 Fixed indendation issues 2023-04-27 02:31:30 +05:30
Prianka Liz Kariat
9c98435027 Updated iOS framework names 2023-04-27 02:15:16 +05:30
MediaPipe Team
c44cc30ece DetectionPostProcessingGraph for post processing raw tensors from detection models.
PiperOrigin-RevId: 527363291
2023-04-26 13:44:54 -07:00
MediaPipe Team
48aa88f39d Change object detector learning rate decay to cosine decay.
PiperOrigin-RevId: 527337105
2023-04-26 12:13:17 -07:00
MediaPipe Team
507ed0d91d Add custom metadata for object detection model with out-of-graph nms.
PiperOrigin-RevId: 527083453
2023-04-25 14:58:51 -07:00
MediaPipe Team
17f5b95387 Internal change.
PiperOrigin-RevId: 527010360
2023-04-25 10:41:52 -07:00
Sebastian Schmidt
9e30b00685 Invoke the FaceStylizer callback even if no faces are detected
PiperOrigin-RevId: 527008261
2023-04-25 10:35:12 -07:00
Jiuqiang Tang
3bc8276678 Remove "All Rights Reserved." in copyright headers.
PiperOrigin-RevId: 526982992
2023-04-25 09:06:29 -07:00
Prianka Liz Kariat
3390325250 Updated documentation 2023-04-25 11:23:38 +05:30
Prianka Liz Kariat
6ac39c9b93 Updated name of common objects pod 2023-04-25 04:26:13 +05:30
Prianka Liz Kariat
d63d3f61d7 Added podspec for CommonObjects and Vision tasks 2023-04-25 04:22:40 +05:30
Prianka Liz Kariat
8b44a7f181 Updated text podspec 2023-04-25 04:22:18 +05:30
Prianka Liz Kariat
472947818e Updated ios cocoapods build script 2023-04-25 04:22:03 +05:30
Prianka Liz Kariat
6eee726025 Updated build rules for iOS frameworks to duplicate symbols 2023-04-25 04:21:41 +05:30
kinaryml
ca5fca1db7 Mark index as unused 2023-04-24 11:26:36 -07:00
kinaryml
b511822815 Removed an unnecessary check and updated tests to check if the masks are generated or not 2023-04-24 11:23:27 -07:00
Sebastian Schmidt
61854dc6a3 Create Pose Detector Web API
PiperOrigin-RevId: 526672533
2023-04-24 09:53:05 -07:00
Jiuqiang Tang
6773188e26 Make FaceLandmarksConnections to be a public class.
PiperOrigin-RevId: 526667505
2023-04-24 09:34:14 -07:00
Sebastian Schmidt
35cf8c35f2 Internal change
PiperOrigin-RevId: 526658482
2023-04-24 09:01:55 -07:00
kinaryml
0b1eb39870 Updated copyright 2023-04-21 11:48:06 -07:00
kinaryml
2a2a55d1b8 Added Language Detector Python API and fixed a typo in Interactive Segmenter Options' docstring 2023-04-21 11:46:21 -07:00
MediaPipe Team
a6c35e9ba5 Fixes the typos in tasks internal files.
PiperOrigin-RevId: 526063515
2023-04-21 09:46:39 -07:00
Sebastian Schmidt
9be748db00 Create MPImage type for Web
PiperOrigin-RevId: 525873209
2023-04-20 16:01:04 -07:00
Sebastian Schmidt
e9bb849503 Fix Typo
PiperOrigin-RevId: 525861968
2023-04-20 15:15:13 -07:00
MediaPipe Team
983932b6dd This will fix the multiple typos in the new tasks internal files
PiperOrigin-RevId: 525788850
2023-04-20 10:43:35 -07:00
Kinar R
9032bce577
Update copyright 2023-04-20 13:21:56 +05:30
Kinar R
21ddba0d60
Update copyright 2023-04-20 13:21:31 +05:30
Copybara-Service
44aa607e06 Merge pull request #4302 from kinaryml:segmenter-python-add-labels
PiperOrigin-RevId: 525571089
2023-04-19 15:46:20 -07:00
Sebastian Schmidt
ffbd799b8d Extract shared types to create and test landmarks
PiperOrigin-RevId: 525568412
2023-04-19 15:37:42 -07:00
MediaPipe Team
5bd3282515 Internal change
PiperOrigin-RevId: 525513792
2023-04-19 12:05:44 -07:00
MediaPipe Team
eb62479190 Internal change
PiperOrigin-RevId: 525487344
2023-04-19 10:35:49 -07:00
Bekzhan Bekbolatuly
9818ebb630 Internal change
PiperOrigin-RevId: 525476655
2023-04-19 09:58:22 -07:00
MediaPipe Team
0aea6d90a8 Internal change
PiperOrigin-RevId: 525407296
2023-04-19 04:10:02 -07:00
MediaPipe Team
de84696be6 Internal change
PiperOrigin-RevId: 525390694
2023-04-19 02:39:54 -07:00
kinaryml
f87ffd92a0 Removed optional for defaults in some tasks and updated various tests to be consistent with that of Pose Landmarker's 2023-04-18 23:28:10 -07:00
kinaryml
00f966655b Fixed a typo in a docstring 2023-04-18 22:48:37 -07:00
kinaryml
1688d0fa79 Added more pose landmarker tests and updated face landmarker tests to cover all the results 2023-04-18 22:45:46 -07:00
Kinar R
39742b6641
Merge branch 'google:master' into pose-landmarker-python 2023-04-19 10:21:23 +05:30
kinaryml
a1aab66c8d Fixed a typo in docstring 2023-04-18 21:50:27 -07:00
kinaryml
67b72e4fe9 Code cleanup 2023-04-18 21:43:38 -07:00
kinaryml
1cb404bea1 Changed labels to be a property 2023-04-18 21:31:14 -07:00
Kinar R
d621df8046
Merge branch 'google:master' into segmenter-python-add-labels 2023-04-19 09:51:42 +05:30
Prianka Liz Kariat
99420d35f3 Update build_ios_framework.sh 2023-04-19 00:25:10 +05:30
Copybara-Service
e15d98298f Merge pull request #4280 from kinaryml:image-segmenter-python-api-updates
PiperOrigin-RevId: 525204893
2023-04-18 11:36:09 -07:00
Prianka Liz Kariat
f75eb57956 Update build_ios_framework.sh 2023-04-18 23:46:05 +05:30
Prianka Liz Kariat
eb0aa5056a Updated documentation 2023-04-18 23:20:11 +05:30
Prianka Liz Kariat
d7c96dea6a Updated formatting of files 2023-04-18 23:10:47 +05:30
Prianka Liz Kariat
24bd7a6b9f Merge branch 'master' into ios-text-cocoapods-force-load 2023-04-18 23:03:56 +05:30
Prianka Liz Kariat
49b2c7c2cc Added iOS task text cocoapods podspec 2023-04-18 23:02:58 +05:30
Prianka Liz Kariat
7ad2b7b32f Added shell script for building cocoapods archive 2023-04-18 23:02:35 +05:30
Prianka Liz Kariat
a774399630 Added targets for iOS text frameworks 2023-04-18 23:02:16 +05:30
MediaPipe Team
64d1e74c20 Internal MediaPipe Tasks change
PiperOrigin-RevId: 525182282
2023-04-18 10:20:03 -07:00
kinaryml
1919b0e341 Updated docstrings for get_labels 2023-04-18 02:54:03 -07:00
kinaryml
723cb2a919 Populate labels using model metadata for the ImageSegmenter Python API 2023-04-18 02:49:13 -07:00
Sebastian Schmidt
47e55fcf2f Add HAND_CONNECTIONS to HandLandmarker and GestureRecognizer
PiperOrigin-RevId: 524951052
2023-04-17 14:31:40 -07:00
MediaPipe Team
2564fec44c Internal MediaPipe Tasks change.
PiperOrigin-RevId: 524942203
2023-04-17 14:09:14 -07:00
Sebastian Schmidt
b147002b7e Support new output format for InteractiveSegmenter
PiperOrigin-RevId: 524940992
2023-04-17 14:04:56 -07:00
Copybara-Service
cd4ae32a10 Merge pull request #4288 from google:pr4279
PiperOrigin-RevId: 524443045
2023-04-14 19:54:49 -07:00
MediaPipe Team
411ffaeb43 Update Java interactive segmenter to output both confidence masks and category mask optionally.
PiperOrigin-RevId: 524442070
2023-04-14 19:45:27 -07:00
Jiuqiang Tang
e14a88052a Remove resizing and rotation from face stylizer's postprocessing step.
The resizing and rotation logic is useful when the pipeline needs to paste the stylized face image back to the original image. To only output the stylized face image, resizing and rotation may result to strange outputs.

PiperOrigin-RevId: 524441372
2023-04-14 19:37:52 -07:00
Copybara-Service
8b3395edfb Merge pull request #4283 from priankakariatyml:ios-timestamp-renaming
PiperOrigin-RevId: 524420908
2023-04-14 17:13:17 -07:00
Jiuqiang Tang
67ed42d9c6 Always link the "pose_landmarks_detector_graph" target.
PiperOrigin-RevId: 524397174
2023-04-14 15:14:20 -07:00
kinaryml
5f5ce22020 Added the PoseLandmarker Python API and a simple test 2023-04-14 14:09:15 -07:00
MediaPipe Team
abd6574c6d Internal MediaPipe Tasks change.
PiperOrigin-RevId: 524377097
2023-04-14 13:49:31 -07:00
Sebastian Schmidt
92f45c98d8 Support new output format for ImageSegmenter
PiperOrigin-RevId: 524371021
2023-04-14 13:25:10 -07:00
MediaPipe Team
f5197a3adc Expose get_graph_config function for task_runner.
PiperOrigin-RevId: 524365682
2023-04-14 13:03:32 -07:00
MediaPipe Team
dbeb5a8126 Add Pose Landmarker Java API
PiperOrigin-RevId: 524359521
2023-04-14 12:36:37 -07:00
Sebastian Schmidt
3f1fc6f520 Add model links to NPM Readme
PiperOrigin-RevId: 524353352
2023-04-14 12:15:28 -07:00
Bekzhan Bekbolatuly
54d208aa5c Internal change
PiperOrigin-RevId: 524345939
2023-04-14 11:44:36 -07:00
MediaPipe Team
257fa01b68 Internal MediaPipe Tasks change.
PiperOrigin-RevId: 524345005
2023-04-14 11:40:18 -07:00
MediaPipe Team
27038f534a Internal change.
PiperOrigin-RevId: 524336899
2023-04-14 11:11:07 -07:00
Jiuqiang Tang
534da98ccb Internal changes
PiperOrigin-RevId: 524334347
2023-04-14 11:01:53 -07:00
Jiuqiang Tang
e6bb9e29c0 Internal change
PiperOrigin-RevId: 524317439
2023-04-14 10:01:01 -07:00
Bekzhan Bekbolatuly
89e6b824ae Update TF version to 2023-04-12
PiperOrigin-RevId: 524301262
2023-04-14 08:49:05 -07:00