def initialise_models():
palm_detector = BlazePalm().to(gpu)
palm_detector.load_weights("models/blazepalm.pth")
palm_detector.load_anchors("models/anchors_palm.npy")
palm_detector.min_score_thresh = .75
hand_regressor = BlazeHandLandmark().to(gpu)
hand_regressor.load_weights("models/blazehand_landmark.pth")
return palm_detector,hand_regressor
back_detector = True
face_detector = BlazeFace(back_model=back_detector).to(gpu)
if back_detector:
face_detector.load_weights("blazefaceback.pth")
face_detector.load_anchors("anchors_face_back.npy")
else:
face_detector.load_weights("blazeface.pth")
face_detector.load_anchors("anchors_face.npy")
palm_detector = BlazePalm().to(gpu)
palm_detector.load_weights("blazepalm.pth")
palm_detector.load_anchors("anchors_palm.npy")
palm_detector.min_score_thresh = .75
hand_regressor = BlazeHandLandmark().to(gpu)
hand_regressor.load_weights("blazehand_landmark.pth")
face_regressor = BlazeFaceLandmark().to(gpu)
face_regressor.load_weights("blazeface_landmark.pth")
WINDOW = 'test'
cv2.namedWindow(WINDOW)
if len(sys.argv) > 1:
capture = cv2.VideoCapture(sys.argv[1])
mirror_img = False
else:
capture = cv2.VideoCapture(2)
mirror_img = True
if capture.isOpened():
back_detector = True
face_detector = BlazeFace(back_model=back_detector).to(gpu)
if back_detector:
face_detector.load_weights("blazefaceback.pth")
face_detector.load_anchors("anchors_face_back.npy")
else:
face_detector.load_weights("blazeface.pth")
face_detector.load_anchors("anchors_face.npy")
palm_detector = BlazePalm().to(gpu)
palm_detector.load_weights("blazepalm.pth")
palm_detector.load_anchors("anchors_palm.npy")
palm_detector.min_score_thresh = .75
hand_regressor = BlazeHandLandmark().to(gpu)
hand_regressor.load_weights("blazehand_landmark.pth")
face_regressor = BlazeFaceLandmark().to(gpu)
face_regressor.load_weights("blazeface_landmark.pth")
WINDOW='test'
cv2.namedWindow(WINDOW)
if len(sys.argv) > 1:
capture = cv2.VideoCapture(sys.argv[1])
mirror_img = False
else:
capture = cv2.VideoCapture(0)
mirror_img = True
back_detector = True
face_detector = BlazeFace(back_model=back_detector).to(gpu)
if back_detector:
face_detector.load_weights("blazefaceback.pth")
face_detector.load_anchors("anchors_face_back.npy")
else:
face_detector.load_weights("blazeface.pth")
face_detector.load_anchors("anchors_face.npy")
palm_detector = BlazePalm().to(gpu)
palm_detector.load_weights("blazepalm.pth")
palm_detector.load_anchors("anchors_palm.npy")
palm_detector.min_score_thresh = .75
hand_regressor = BlazeHandLandmark().to(gpu)
hand_regressor.load_weights("blazehand_landmark.pth")
face_regressor = BlazeFaceLandmark().to(gpu)
face_regressor.load_weights("blazeface_landmark.pth")
WINDOW = 'test'
cv2.namedWindow(WINDOW)
if len(sys.argv) > 1:
capture = cv2.VideoCapture(sys.argv[1])
mirror_img = False
else:
capture = cv2.VideoCapture(0)
mirror_img = True
if capture.isOpened():
def load_landmark_regressor(self):
hand_regressor = BlazeHandLandmark()
hand_regressor.load_weights(self.landmarks_weights_path)
return hand_regressor