Include object_detector_metadata_schema.fbs and image_segmenter_metadata_schema.fbs
PiperOrigin-RevId: 530769920
This commit is contained in:
parent
f95a782399
commit
e391c76433
6
setup.py
6
setup.py
|
@ -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))
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user