dsfd_error_01_widerface_val.py

# Line 61

def detect_face(image, shrink):
    x = image
    if shrink != 1:
        x = cv2.resize(image, None, None, fx=shrink, fy=shrink, interpolation=cv2.INTER_LINEAR)
    #print('shrink:{}'.format(shrink))
    width = x.shape[1]
    height = x.shape[0]
    x = x.astype(np.float32)
    x -= np.array([104, 117, 123],dtype=np.float32)
    
    x = torch.from_numpy(x).permute(2, 0, 1)
    x = x.unsqueeze(0)
    with torch.no_grad(): # <----------
        x = Variable(x.cuda()) # <----------