refactor for triton inference server

README.md

@@ -1,3 +1,6 @@
 ---
 license: openrail
 ---
+This model repository is meant to be imported into the Triton Inference Server container. The model repository contains the following models:
+1. Catnet
+2. DTD

catnet/config.pbtxt

@@ -0,0 +1,31 @@
+name: "catnet"
+platform: "onnxruntime_onnx"
+input [
+  {
+    name: "input_image"
+    data_type: TYPE_FP32
+    dims: [ 1, 24, -1, -1 ]
+  },
+  {
+    name: "input_label"
+    data_type: TYPE_INT64
+    dims: [ 1, -1, -1 ]
+  },
+  {
+    name: "input_qtable"
+    data_type: TYPE_FP32
+    dims: [ 1, 1, 8, 8 ]
+  }
+]
+output [
+  {
+    name: "out_1"
+    data_type: TYPE_FP32
+    dims: [ 1 ]
+  },
+  {
+    name: "output_pred"
+    data_type: TYPE_FP32
+    dims: [ -1, 2, -1, -1 ]
+  }
+]

dtd.onnx → dtd/1/model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a9f525b902a5f87735d2c3df4aa8245f1b10843c4d694491c6c8578d6db928e3
-size 286141885
+oid sha256:cbfe1204d8b0acf34a37ca61d85af15b4c8ee761500ff52895e09bc84d3f1dbf
+size 281497619

dtd/config.pbtxt

@@ -0,0 +1,26 @@
+name: "dtd"
+platform: "onnxruntime_onnx"
+input [
+  {
+    name: "input"
+    data_type: TYPE_FP32
+    dims: [ 1, 3, 512, 512 ]
+  },
+  {
+    name: "dct"
+    data_type: TYPE_INT32
+    dims: [ 1, 512, 512 ]
+  },
+  {
+    name: "qt"
+    data_type: TYPE_INT64
+    dims: [ 1, 1, 8, 8 ]
+  }
+]
+output [
+  {
+    name: "output"
+    data_type: TYPE_FP32
+    dims: [ 1, 2, 512, 512 ]
+  }
+]