first push

utils/monitor.py

@@ -0,0 +1,61 @@
+import os
+import json
+import h5py
+import numpy as np
+from utils.config import cfg
+from cirtorch.networks.imageretrievalnet import init_network, extract_vectors
+from utils.tools import createNet,ManagingFeature 
+
+class Moniting:
+    def __init__(self, barcode = ''):
+        self.barcode = barcode
+        self.jsonpath = cfg.MONITORPATH
+        self.MF = ManagingFeature()
+        if not os.path.exists(self.jsonpath):
+            jsontext = {"monitor":[]}
+            jsondata = json.dumps(jsontext)
+            f = open(self.jsonpath, 'w+', encoding='utf-8')
+            f.write(jsondata)
+            f.close()
+
+    def add(self):
+        with open(self.jsonpath, 'r', encoding='utf-8') as add_f:
+            jsonfile = json.load(add_f)
+        add_f.close()
+        data = list(set(jsonfile['monitor']+self.barcode))
+        jsonfile['monitor'] = data
+        with open(self.jsonpath, 'w') as add_f:
+            json.dump(jsonfile, add_f)
+        add_f.close()
+
+    def search(self):
+        with open(self.jsonpath, 'r', encoding='utf-8') as f:
+            jsonfile = json.load(f)
+            f.close()
+        data = set(jsonfile['monitor'])
+        if self.barcode in data:
+            return 'success'
+        else:
+            return 'nomatch'
+
+    def update(self, net, transform, ms):#update monitor.json
+        Dict = {}
+        path = []
+        collectbarcode = set()
+        for name in os.listdir(cfg.IMG_DIR_TOTAL):
+            barcode = name.split('_')[-1].split('.')[0]
+            path.append(os.sep.join([cfg.IMG_DIR_TOTAL, name]))
+            collectbarcode.add(barcode)
+        vecs, img_paths = extract_vectors(net, path, cfg.RESIZE, transform, ms=ms)
+        data = list(vecs.detach().cpu().numpy().T)
+        for code, feature in zip(img_paths, data):
+            barcode = code.split('_')[-1].split('.')[0]
+            feature = feature.tolist()
+            self.MF.addfeature(barcode, feature)
+        Moniting(list(collectbarcode)).add()
+
+if __name__ == '__main__':
+    barcode = '1'
+    mo = Moniting(barcode)
+    print(mo.search())
+    mo.add(barcode)