acnet.patch

diff -x .git -x .gitignore -x '*.DS_Store' -r -N -r -u ./to_onnx.py esanet/src/models/external_code/ACNet/to_onnx.py
--- ./to_onnx.py	1970-01-01 01:00:00.000000000 +0100
+++ esanet/src/models/external_code/ACNet/to_onnx.py	2020-12-06 11:45:33.000000000 +0100
@@ -0,0 +1,32 @@
+import os
+
+import torch
+
+from ACNet_models_V1 import ACNet
+
+
+if __name__ == '__main__':
+    H = 480
+    W = 640
+    N_CLASSES = 40
+
+    rgb = torch.rand(size=(1, 3, H, W), dtype=torch.float32)
+    depth = torch.rand(size=(1, 1, H, W), dtype=torch.float32)
+
+    out_dir = '../onnx_models'
+    os.makedirs(out_dir, exist_ok=True)
+    onnx_file_path_acnet = os.path.join(out_dir, 'acnet.onnx')
+
+    acnet = ACNet(num_class=N_CLASSES)
+    acnet.eval()
+
+    torch.onnx.export(acnet,
+                      (rgb, depth),
+                      onnx_file_path_acnet,
+                      export_params=True,
+                      input_names=['rgb', 'depth'],
+                      output_names=['output'],
+                      do_constant_folding=True,
+                      verbose=False,
+                      opset_version=10
+                      )