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

Included CPU binary graphs in setup.py

Browse Source
This commit is contained in:
Kinar R 2023-05-18 11:00:12 +05:30 committed by GitHub
parent 45addac249
commit 126df20658
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
setup.py
View File

@ -246,18 +246,18 @@ class BuildModules(build_ext.build_ext):
self._download_external_file(external_file)
# CPU binary graphs
# binary_graphs = [
# 'face_detection/face_detection_short_range_cpu',
# 'face_detection/face_detection_full_range_cpu',
# 'face_landmark/face_landmark_front_cpu',
# 'hand_landmark/hand_landmark_tracking_cpu',
# 'holistic_landmark/holistic_landmark_cpu', 'objectron/objectron_cpu',
# 'pose_landmark/pose_landmark_cpu',
# 'selfie_segmentation/selfie_segmentation_cpu'
# ]
binary_graphs = [
'face_detection/face_detection_short_range_cpu',
'face_detection/face_detection_full_range_cpu',
'face_landmark/face_landmark_front_cpu',
'hand_landmark/hand_landmark_tracking_cpu',
'holistic_landmark/holistic_landmark_cpu', 'objectron/objectron_cpu',
'pose_landmark/pose_landmark_cpu',
'selfie_segmentation/selfie_segmentation_cpu'
]
# GPU binary graphs
binary_graphs = [
binary_graphs += [
'face_detection/face_detection_short_range_gpu',
'face_detection/face_detection_full_range_gpu',
'face_landmark/face_landmark_front_gpu',