Included CPU binary graphs in setup.py

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

View File

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