test

requirements.txt

@@ -1,7 +1,7 @@
 torch
 torchvision
 torchaudio
-numpy
+numpy==1.23.5
 face_alignment
 imageio
 imageio-ffmpeg

src/face3d/models/arcface_torch/torch2onnx.py

@@ -6,7 +6,7 @@ import torch
 def convert_onnx(net, path_module, output, opset=11, simplify=False):
     assert isinstance(net, torch.nn.Module)
     img = np.random.randint(0, 255, size=(112, 112, 3), dtype=np.int32)
-    img = img.astype(np.float64)
+    img = img.astype(np.float)
     img = (img / 255. - 0.5) / 0.5  # torch style norm
     img = img.transpose((2, 0, 1))
     img = torch.from_numpy(img).unsqueeze(0).float()