From 126df20658c11e6ef97f7c64bd781b9b3d768e9b Mon Sep 17 00:00:00 2001 From: Kinar R <42828719+kinaryml@users.noreply.github.com> Date: Thu, 18 May 2023 11:00:12 +0530 Subject: [PATCH] Included CPU binary graphs in setup.py --- setup.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/setup.py b/setup.py index 0712a95b0..ce162b594 100644 --- a/setup.py +++ b/setup.py @@ -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',