Upload 14 files

.gitattributes

@@ -22,3 +22,11 @@ ckpt/model_float16.tflite filter=lfs diff=lfs merge=lfs -text
 ckpt/model_float32.tflite filter=lfs diff=lfs merge=lfs -text
 ckpt/model5x50.onnx filter=lfs diff=lfs merge=lfs -text
 ckpt/saved_model.pb filter=lfs diff=lfs merge=lfs -text
+20x/debug.bin filter=lfs diff=lfs merge=lfs -text
+20x/debug2.bin filter=lfs diff=lfs merge=lfs -text
+20x/modelpnnx20x_float.pt filter=lfs diff=lfs merge=lfs -text
+20x/modelpnnx20x.ncnn.bin filter=lfs diff=lfs merge=lfs -text
+20x/modelpnnx20x.pnnx.bin filter=lfs diff=lfs merge=lfs -text
+20x/modelpnnx20x.pnnx.onnx filter=lfs diff=lfs merge=lfs -text
+20x/modelpnnx20x.pt filter=lfs diff=lfs merge=lfs -text
+20x/modelpnnx50x.pt filter=lfs diff=lfs merge=lfs -text

20x/debug.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3224be077230488f38918a022fa31634f6c69e33d9e00d739f5b097a63b4b4fc
+size 213434872

20x/debug2.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:83bd07981deb80493d5d291c03cbe86b7a45868672fc94be669faad4044b8772
+size 213401460

20x/modelpnnx20x.ncnn.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0b8f14d19799c39fce865e7637fd96ebe9e5a361fe95f89112dfd22bba949f4a
+size 133294028

20x/modelpnnx20x.pnnx.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b7e6b65006b94e2ccec8fbaa268cd14492b82aecb5ab53d2cd97ebef52fd3f77
+size 221641234

20x/modelpnnx20x.pnnx.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bbe2b8a3e9f7da5275209c554921343b1c128d981f3a6fc040bcd890ba6da0ad
+size 111237962

20x/modelpnnx20x.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:507cafbcef26b2e0cdd00c6ae7709e65a9b5a8dfa83b8cae7043e93409f1c25c
+size 151706608

20x/modelpnnx20x_float.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:67b228fef8cc7ac8cce4d77b41c774e7f6df733a5d8eabf9965ad9a88608695e
+size 151716342

20x/modelpnnx20x_ncnn.py

@@ -0,0 +1,31 @@
+import numpy as np
+import ncnn
+import torch
+
+def test_inference():
+    torch.manual_seed(0)
+    in0 = torch.rand(1, 3, 384, 384, dtype=torch.float)
+    in1 = torch.rand(1, 3, 384, 384, dtype=torch.float)
+    out = []
+
+    with ncnn.Net() as net:
+         net.load_param("/Users/raoulritter/STB-VMM/20x/modelpnnx20x.ncnn.param")
+         net.load_model("/Users/raoulritter/STB-VMM/20x/modelpnnx20x.ncnn.bin")
+
+         with net.create_extractor() as ex:
+            ex.input("in0", ncnn.Mat(in0.squeeze(0).numpy()).clone())
+            ex.input("in1", ncnn.Mat(in1.squeeze(0).numpy()).clone())
+
+            _, out0 = ex.extract("out0")
+            out.append(torch.from_numpy(np.array(out0)).unsqueeze(0))
+            _, out1 = ex.extract("out1")
+            out.append(torch.from_numpy(np.array(out1)).unsqueeze(0))
+            _, out2 = ex.extract("out2")
+            out.append(torch.from_numpy(np.array(out2)).unsqueeze(0))
+            _, out3 = ex.extract("out3")
+            out.append(torch.from_numpy(np.array(out3)))
+
+    if len(out) == 1:
+        return out[0]
+    else:
+        return tuple(out)

20x/modelpnnx50x.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aa56f56688b0cca51e5d7021717f0c7dc3b82caba7b5c51689aaecf220f5b037
+size 151689558