np.squeeze expects int or tuple
see:https://docs.scipy.org/doc/numpy/reference/generated/numpy.squeeze.html
This commit is contained in:
Родитель
138ff7399d
Коммит
b8998d7f5a
6
demo.py
6
demo.py
|
@ -61,9 +61,9 @@ def demo():
|
|||
|
||||
# reshape outputs
|
||||
input_img = np.squeeze(input_img)
|
||||
shape = np.squeeze(face_shape,[0])
|
||||
color = np.squeeze(face_color,[0])
|
||||
landmarks_2d = np.squeeze(landmarks_2d,[0])
|
||||
shape = np.squeeze(face_shape, (0))
|
||||
color = np.squeeze(face_color, (0))
|
||||
landmarks_2d = np.squeeze(landmarks_2d, (0))
|
||||
|
||||
# save output files
|
||||
# cropped image, which is the direct input to our R-Net
|
||||
|
|
Загрузка…
Ссылка в новой задаче