Commit Graph

429 Commits

Author SHA1 Message Date
MediaPipe Team
8b2c937b9e Migrate AudioClassifier C++ to use new ClassificationResult struct.
PiperOrigin-RevId: 486162683
2022-11-04 09:46:09 -07:00
MediaPipe Team
62cd67e996 Internal change
PiperOrigin-RevId: 486148328
2022-11-04 08:42:39 -07:00
Jiuqiang Tang
5f5f50d8f7 Implement MediaPipe Tasks Python AudioData.
PiperOrigin-RevId: 486147173
2022-11-04 08:32:30 -07:00
Copybara-Service
5024c815f1 Merge pull request #3813 from kinaryml:image-segmenter-minor-fixes
PiperOrigin-RevId: 486051659
2022-11-03 21:41:37 -07:00
MediaPipe Team
b2cc2cc60c Internal changes
PiperOrigin-RevId: 486014305
2022-11-03 17:23:20 -07:00
MediaPipe Team
06cb73fc81 Modified internal dependencies.
PiperOrigin-RevId: 485992876
2022-11-03 15:44:36 -07:00
kinaryml
e2d50745ac Revised image embedder implementation 2022-11-03 14:30:21 -07:00
MediaPipe Team
d29c3d7512 Add metadata writer to image_classifier in model_maker
PiperOrigin-RevId: 485926985
2022-11-03 11:30:38 -07:00
kinaryml
492607152a Reverted changes to rect 2022-11-03 10:56:38 -07:00
Kinar R
5a68ba84b6
Merge branch 'master' into image-embedder-python 2022-11-03 23:24:31 +05:30
Jiuqiang Tang
b472d8ff66 Add the missing __init__.py file.
PiperOrigin-RevId: 485892501
2022-11-03 09:24:12 -07:00
Copybara-Service
716e59f90c Merge pull request #3801 from kinaryml:gesture-recognizer-python
PiperOrigin-RevId: 485884796
2022-11-03 08:52:17 -07:00
kinaryml
14e18a03d6 Removed unused custom classifier options in tests 2022-11-03 02:54:57 -07:00
kinaryml
de619e2702 Removed a comment 2022-11-03 02:33:50 -07:00
kinaryml
2001e2b77c Removed unused file 2022-11-03 02:32:55 -07:00
kinaryml
6e9a070dd1 Added text classifier implementation and tests 2022-11-03 02:29:40 -07:00
kinaryml
35a04522fa Moved the OutputType and Activation classes to ImageSegmenter's inner classes 2022-11-03 00:46:00 -07:00
MediaPipe Team
ddf37d014e Fix test utils bug when two file names have same ending.
PiperOrigin-RevId: 485780917
2022-11-02 22:33:36 -07:00
Jiuqiang Tang
ac6aa9ce10 Fix a bug in the AndroidPacketCreator's createImage method.
PiperOrigin-RevId: 485716828
2022-11-02 15:56:49 -07:00
Jiuqiang Tang
01d87f0867 Expose MediaPipe Tasks in MediaPipe PyPI packages.
PiperOrigin-RevId: 485690163
2022-11-02 14:07:47 -07:00
MediaPipe Team
0bf5110a7e Add a model_maker internal library directory.
PiperOrigin-RevId: 485665201
2022-11-02 12:31:58 -07:00
MediaPipe Team
d61ab92b90 Add classification result structs and use then in ImageClassifier & TextClassifier.
PiperOrigin-RevId: 485567133
2022-11-02 05:13:12 -07:00
MediaPipe Team
aab5f84aae Internal change
PiperOrigin-RevId: 485553891
2022-11-02 03:49:48 -07:00
MediaPipe Team
475e6b4fd5 Modify ClassificationAggregationCalculator to output new unified formats.
PiperOrigin-RevId: 485546547
2022-11-02 03:11:20 -07:00
MediaPipe Team
f1f123d255 Extracted common logics from the ImageToTensorCalculator such that it can be
reused by other calculators.

PiperOrigin-RevId: 485472451
2022-11-01 18:43:46 -07:00
MediaPipe Team
aaf98ea43c Add CPU invoke to Perfetto trace
PiperOrigin-RevId: 485464221
2022-11-01 17:50:39 -07:00
MediaPipe Team
ecd93ff080 Internal changes
PiperOrigin-RevId: 485445138
2022-11-01 16:18:06 -07:00
Kinar R
3a2f30185f
Merge branch 'master' into gesture-recognizer-python 2022-11-02 04:11:18 +05:30
kinaryml
c5765ac836 Refactored Rect to use top-left coordinates and appropriately updated the Image Classifier and Gesture Recognizer APIs/tests 2022-11-01 15:37:00 -07:00
Yuqi Li
e719f7b4e2 Model Maker Gesture Recognizer: add metadata writer and create model bundle.
PiperOrigin-RevId: 485426865
2022-11-01 15:02:45 -07:00
MediaPipe Team
c6a64683f6 Add ConcatenateClassificationListCalculator
PiperOrigin-RevId: 485398597
2022-11-01 13:13:48 -07:00
MediaPipe Team
700971de70 Internal change
PiperOrigin-RevId: 485396135
2022-11-01 13:05:43 -07:00
Copybara-Service
6e0397b226 Merge pull request #3800 from kinaryml:python-test-proto-equals
PiperOrigin-RevId: 485340924
2022-11-01 09:42:19 -07:00
Frederic Rechtenstein
60a19f4fe2 Fix missing space in docstring.
PiperOrigin-RevId: 485337309
2022-11-01 09:27:29 -07:00
Jiuqiang Tang
e4899c8207 MediaPipe Python audio base layer.
PiperOrigin-RevId: 485329470
2022-11-01 08:53:36 -07:00
kinaryml
a913255080 Removed min score thres from tests 2022-10-31 23:07:05 -07:00
kinaryml
d3b472e888 Add allow_list/deny_list support 2022-10-31 22:16:37 -07:00
Kinar R
1aaaca1e12
Merge branch 'google:master' into gesture-recognizer-python 2022-11-01 08:16:43 +05:30
Jiuqiang Tang
86951e048d Remove the unused TaskRunner from the mediapipe tasks python code.
PiperOrigin-RevId: 485181210
2022-10-31 16:54:30 -07:00
MediaPipe Team
9f8b5e5c11 Add allow_list/deny_list to gesture recognizer options.
PiperOrigin-RevId: 485141209
2022-10-31 13:57:06 -07:00
Jiuqiang Tang
4717ac298c Add EndLoopMatrixCalculator
PiperOrigin-RevId: 485097386
2022-10-31 11:08:21 -07:00
kinaryml
2b5a077579 Updated comments 2022-10-31 05:48:45 -07:00
kinaryml
d635b4281e Added a test for the canned classification of the gesture victory 2022-10-31 05:47:28 -07:00
kinaryml
888ddd4b74 Removed unused classifier options proto 2022-10-31 05:37:24 -07:00
kinaryml
19be9e9012 Revised gesture recognizer implementation 2022-10-31 05:34:31 -07:00
Kinar R
5ec87c8bd2
Merge branch 'master' into gesture-recognizer-python 2022-10-31 16:47:43 +05:30
MediaPipe Team
459214e6a3 Rename index_by_label to label_names.
PiperOrigin-RevId: 484956259
2022-10-30 22:26:17 -07:00
Copybara-Service
7bcf322625 Merge pull request #3739 from kinaryml:image-segmenter-python-impl
PiperOrigin-RevId: 484922757
2022-10-30 17:11:42 -07:00
kinaryml
fb4872b068 Refactored code and removed some issues 2022-10-30 15:42:26 -07:00
MediaPipe Team
be13a76928 Integrate canned and custom gesture classifier.
PiperOrigin-RevId: 484914164
2022-10-30 15:36:09 -07:00
kinaryml
4b66599419 Updated docstring in gesture_recognizer 2022-10-30 09:10:15 -07:00
Kinar R
6f485ae3dd
Merge branch 'google:master' into gesture-recognizer-python 2022-10-30 20:55:53 +05:30
kinaryml
f62cfd1690 Removed classification proto to use the existing category dataclass instead and removed NormalizedLandmarkList and LandmarkList dataclasses 2022-10-30 08:23:14 -07:00
Hadon Nash
313b387ced Adding SwitchContainer option "tick_input_stream" to specify which input streams indicate the output timestamps to expect from the contained graphs.
PiperOrigin-RevId: 484615286
2022-10-28 14:04:20 -07:00
Jiuqiang Tang
e16be2e8fa Implement the initial version of TensorsToAudioCalculator that supports ifft and inverse hann windowing.
PiperOrigin-RevId: 484605092
2022-10-28 13:19:43 -07:00
Hadon Nash
c5c639d634 Internal change
PiperOrigin-RevId: 484583911
2022-10-28 11:46:10 -07:00
Jiuqiang Tang
f4f8b11ffc Fix the comment.
PiperOrigin-RevId: 484571254
2022-10-28 10:57:13 -07:00
Kinar R
334f641463
Merge branch 'master' into image-segmenter-python-impl 2022-10-28 22:50:33 +05:30
Kinar R
0f7c5d5e90
Merge branch 'google:master' into gesture-recognizer-python 2022-10-28 14:08:52 +05:30
kinaryml
b81b5a9035 Added a test for min_gesture_confidence 2022-10-28 01:38:15 -07:00
MediaPipe Team
de5fe27e05 Modified internal dependencies.
PiperOrigin-RevId: 484407262
2022-10-27 19:08:19 -07:00
MediaPipe Team
abd1ff66c8 Fix https://github.com/google/mediapipe/issues/3784
PiperOrigin-RevId: 484365654
2022-10-27 15:32:44 -07:00
MediaPipe Team
fc1d75cc99 Add CombinedPredictionCalculator.
PiperOrigin-RevId: 484301880
2022-10-27 11:16:12 -07:00
MediaPipe Team
ee84e447b2 Internal change
PiperOrigin-RevId: 484299808
2022-10-27 11:08:31 -07:00
kinaryml
8194513934 Updated docstrings and removed the redundant private helper method 2022-10-26 23:56:54 -07:00
Jiuqiang Tang
a941c5cdd1 Create MediaPipe "tasks-text" AAR.
PiperOrigin-RevId: 484004494
2022-10-26 10:18:22 -07:00
MediaPipe Team
254f786624 Add an option to set image preprocessing backend as gpu.
PiperOrigin-RevId: 483888202
2022-10-26 00:46:19 -07:00
kinaryml
f8a98ccba4 Added a test utility method to compare protos directly 2022-10-25 23:38:32 -07:00
kinaryml
fbf7ba6f1a Reverted some changes to rect 2022-10-25 23:15:16 -07:00
kinaryml
75af46d273 Revised API to align with recent changes 2022-10-25 23:13:12 -07:00
Kinar R
0de97497fa
Merge branch 'master' into gesture-recognizer-python 2022-10-26 11:37:12 +05:30
Copybara-Service
ae5b09e2b2 Merge pull request #3738 from kinaryml:image-classification-python-impl
PiperOrigin-RevId: 483818404
2022-10-25 17:26:32 -07:00
MediaPipe Team
a28c9d2c26 Set steps_per_epoch to None when calling model.fit() method for image classifier.
PiperOrigin-RevId: 483764377
2022-10-25 13:44:03 -07:00
MediaPipe Team
36bd9abb8f Internal change
PiperOrigin-RevId: 483751427
2022-10-25 12:53:39 -07:00
MediaPipe Team
21abfc9125 Update gpu origin.
PiperOrigin-RevId: 483742652
2022-10-25 12:14:28 -07:00
kinaryml
8762d15c81 Added remaining tests for the GestureRecognizer Python MediaPipe Tasks API 2022-10-25 11:11:15 -07:00
MediaPipe Team
d240c009e2 Remove unnecessary location_data_proto dependency on rect_proto.
PiperOrigin-RevId: 483679555
2022-10-25 08:24:07 -07:00
kinaryml
18eb089d39 Added a simple test to verify gesture recognition results 2022-10-25 07:38:04 -07:00
MediaPipe Team
6b0a7fb281 Reverting back to special handling for Egl Thread Exit on Android
PiperOrigin-RevId: 483505151
2022-10-24 15:45:35 -07:00
MediaPipe Team
94cd134809 Add support for image rotation in Java vision tasks.
PiperOrigin-RevId: 483493729
2022-10-24 15:01:23 -07:00
MediaPipe Team
3d88b1797a Switch CHECK to a status in resource handling code. Expand error message.
PiperOrigin-RevId: 483438131
2022-10-24 11:27:15 -07:00
Jiuqiang Tang
2cf9523468 Fix the java_package name.
PiperOrigin-RevId: 483428848
2022-10-24 10:55:16 -07:00
MediaPipe Team
2f2baeff68 Add support for rotation in ImageEmbedder & ImageSegmenter C++ APIs
PiperOrigin-RevId: 483416498
2022-10-24 10:14:05 -07:00
MediaPipe Team
0fd69e8d83 Open-source some tokenizer unit tests.
PiperOrigin-RevId: 483399326
2022-10-24 09:12:52 -07:00
kinaryml
9a1a9d4c13 Added files needed for the GestureRecognizer API implementation 2022-10-24 06:08:27 -07:00
MediaPipe Team
af051dcb62 internal change
PiperOrigin-RevId: 483308781
2022-10-24 01:39:23 -07:00
Jiuqiang Tang
de93d06f87 Implement build rules and targets to create MediaPipe Tasks AARs
PiperOrigin-RevId: 483297891
2022-10-24 00:45:32 -07:00
Yuqi Li
ab17be9294 Metadata Writer: Add Metadata Writer for image classifier.
PiperOrigin-RevId: 483282627
2022-10-23 23:05:32 -07:00
MediaPipe Team
ec2a34d2a4 Replace pytype_struct_contrib_test by py_strict_test. Also remove unnecessary BUILD attributes.
PiperOrigin-RevId: 483237371
2022-10-23 16:45:02 -07:00
MediaPipe Team
ab0193bbb7 Internal change
PiperOrigin-RevId: 483078695
2022-10-22 15:55:46 -07:00
kinaryml
ebb2686fb4 Moved OutputType and Activation to image_segmenter 2022-10-22 03:34:26 -07:00
MediaPipe Team
d8006a2f87 Use model bundle for gesture recognizer.
PiperOrigin-RevId: 482960305
2022-10-21 21:54:34 -07:00
MediaPipe Team
404323f631 Add Mediapipe Tasks Gesture Recognizer benchmarks
PiperOrigin-RevId: 482935780
2022-10-21 18:33:49 -07:00
Jiuqiang Tang
abed54ea30 Rename the mediapipe java image container from Image to MPImage.
PiperOrigin-RevId: 482933122
2022-10-21 18:17:35 -07:00
MediaPipe Team
7196db275e Internal change
PiperOrigin-RevId: 482925717
2022-10-21 17:28:49 -07:00
MediaPipe Team
ea1d85d811 Update model_task_graph to support multiple local model resources.
PiperOrigin-RevId: 482917453
2022-10-21 16:42:58 -07:00
MediaPipe Team
4a6c23a76a Internal change
PiperOrigin-RevId: 482906478
2022-10-21 15:48:54 -07:00
MediaPipe Team
d0437b7f91 Add tensor_index and tensor_name fields to ClassificationList
PiperOrigin-RevId: 482901854
2022-10-21 15:27:35 -07:00
kinaryml
024a6866a7 Removed some unneeded aliases 2022-10-21 13:39:59 -07:00
MediaPipe Team
36d69971a7 Internal change
PiperOrigin-RevId: 482875698
2022-10-21 13:36:41 -07:00