update padding

Author: liuwei9

py/paddingMode.py

@@ -0,0 +1,52 @@
+import torch.nn as nn
+import torch
+import numpy as np
+import json
+import sys
+
+
+def main(cfgPath):
+    with open(cfgPath) as f:
+        a = json.load(f)
+    channel = a['padding']['channelIn']
+    row = a['padding']['rowNumIn']
+    col = a['padding']['colNumIn']
+    enPadding = a['padding']['enPadding']
+    src = a['padding']['src_py']
+    dst = a['padding']['dst_py']
+    computeChannel = a['padding']['COMPUTE_CHANNEL_NUM']
+    zeroData = a['padding']['zeroDara']
+    zeroNum = a['padding']['zeroNum']
+
+    x = torch.randint(0, 127, (channel, row, col))
+    xx = np.array(x)
+    y = x
+    if enPadding:
+        pad = nn.ConstantPad2d(padding=(zeroNum, zeroNum, zeroNum, zeroNum), value=zeroData)
+        y = pad(x)
+    yy = np.array(y)
+    fp1 = open(src, 'w')
+    fp2 = open(dst, 'w')
+
+    def save_xx(high, width, channel, xx, fp):
+        out = []
+        for h in range(high):
+            for w in range(width):
+                for c in range(channel):
+                    out.append(xx[c][h][w])
+                    if len(out) == computeChannel:
+                        out.reverse()
+                        for m in out:
+                            m = m.item()
+                            fp.write('%02x' % m)
+                        fp.write('\n')
+                        out = []
+        fp.close()
+
+    save_xx(xx.shape[1], xx.shape[2], xx.shape[0], xx, fp1)
+    save_xx(yy.shape[1], yy.shape[2], yy.shape[0], yy, fp2)
+
+
+if __name__ == '__main__':
+    # print(sys)
+    main("../simData/config.json")

simData/config.json

@@ -0,0 +1,17 @@
+{
+  "padding": {
+    "enPadding": true,
+    "channelIn": 8,
+    "rowNumIn": 640,
+    "colNumIn": 320,
+    "zeroDara": 8,
+    "zeroNum": 2,
+    "COMPUTE_CHANNEL_NUM": 8,
+    "DATA_WIDTH": 8,
+    "CHANNEL_WIDTH": 10,
+    "FEATURE_WIDTH": 11,
+    "src_py": "G:/SpinalStudy/simData/src.txt",
+    "dst_py": "G:/SpinalStudy/simData/dst.txt",
+    "dst_scala": "G:/SpinalStudy/simData/dst_scala.txt"
+  }
+}