cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Include object_detector_metadata_schema.fbs and image_segmenter_metadata_schema.fbs

Browse Source 
PiperOrigin-RevId: 530769920
This commit is contained in:
MediaPipe Team 2023-05-09 18:42:28 -07:00 committed by Copybara-Service
parent f95a782399
commit e391c76433
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
setup.py
View File

@ -313,7 +313,11 @@ class GenerateMetadataSchema(build_ext.build_ext):
_copy_to_build_lib_dir(
self.build_lib,
'mediapipe/tasks/metadata/' + target + '_generated.py')
schema_file = 'mediapipe/tasks/metadata/metadata_schema.fbs'
for schema_file in [
'mediapipe/tasks/metadata/metadata_schema.fbs',
'mediapipe/tasks/metadata/object_detector_metadata_schema.fbs',
'mediapipe/tasks/metadata/image_segmenter_metadata_schema.fbs',
]:
shutil.copyfile(schema_file,
os.path.join(self.build_lib + '/', schema_file))