Upload 7 files

ensemble.py

@@ -0,0 +1,56 @@
+#coding:utf-8
+import os
+import pickle
+import zipfile
+from os.path import basename
+import tqdm
+import numpy as np
+
+def read_numpy(file_path):
+    return np.load(file_path)
+
+def load_pkl(file_path):
+    with open(file_path, 'rb') as f:
+        data = pickle.load(f)
+        f.close()
+    return data
+
+with open('../data/label_map.txt', 'r') as f:
+    ids = [line.strip() for line in f.readlines()]
+
+def get_prob_labels(file_path):
+    pred_meta = load_pkl(file_path)
+    out_probs = [x['pred_score'].cpu().numpy() for x in pred_meta]
+    out_probs = np.vstack(out_probs) # [51440, 3215]
+    out_labels = np.argmax(out_probs, axis=1) # [51440]
+    return out_probs, out_labels
+
+# RGB: 
+vswin_small_rgb_prob, _ = get_prob_labels('./vswin_small_rgb/result_swin_small_rgb.pkl')
+vswin_base_rgb_prob, _ = get_prob_labels('./vswin_base_rgb/result_swin_base_rgb.pkl')
+vswin_large_rgb_prob, _ = get_prob_labels('./vswin_large_rgb/result_swin_large_rgb.pkl')
+rgb_prob = 0.4 * vswin_large_rgb_prob + 0.4 * vswin_base_rgb_prob + 0.2 * vswin_small_rgb_prob
+
+# Depth: 
+vswin_small_depth_prob, _ = get_prob_labels('./vswin_small_depth/result_swin_small_depth.pkl')
+vswin_base_depth_prob, _ = get_prob_labels('./vswin_base_depth/result_swin_base_depth.pkl')
+vswin_large_depth_prob, _ = get_prob_labels('./vswin_large_depth/result_swin_large_depth.pkl')
+
+depth_prob = 0.65 * vswin_large_depth_prob + 0.35 * vswin_base_depth_prob
+
+fused_prob = rgb_prob * 0.5 + depth_prob * 0.5  # [51440. 3215]
+
+out_labels = np.argmax(fused_prob, axis=1) # [51440]
+############################################
+
+with open('./answer.txt', 'w') as f:
+    for label in out_labels:
+        f.write(ids[label]+'\n')
+    f.close()
+
+csv_file_path = './answer.txt'
+zip_file_path = './answer.zip'
+
+with zipfile.ZipFile(zip_file_path, 'w') as zipf:
+    zipf.write(csv_file_path, os.path.basename(csv_file_path))
+

vswin_base_depth/result_swin_base_depth.pkl

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

vswin_base_rgb/result_swin-base_rgb.pkl

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

vswin_large_depth/result_swin_large_depth.pkl

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

vswin_large_rgb/result_swin_large_rgb.pkl

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

vswin_small_depth/result_swin_small_depth.pkl

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

vswin_small_rgb/result_swin_small_rgb.pkl

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