Included CPU binary graphs in setup.py
This commit is contained in:
parent
45addac249
commit
126df20658
20
setup.py
20
setup.py
|
@ -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',
|
||||||
|
|
Loading…
Reference in New Issue
Block a user