first push

updatefile.py

@@ -0,0 +1,73 @@
+import sys
+import argparse
+import requests
+from utils.tools import EvaluteMap
+from utils.retrieval_feature import AntiFraudFeatureDataset
+from utils.tools import createNet, rotate_bound
+from utils.monitor import Moniting
+from cirtorch.networks.imageretrievalnet import init_network, extract_vectors
+from utils.config import cfg
+from utils.updateObs import *
+import numpy as np
+import cv2
+import json
+import os, base64, stat, shutil
+import pdb
+import socket
+
+sys.path.append('RAFT')
+sys.path.append('RAFT/core')
+sys.path.append('RAFT/core/utils')
+from RAFT.analysis_video import *
+
+class update:
+    def __init__(self):
+        self.net, self.transform, self.ms = createNet()
+        self.updateMonitor = Moniting()
+
+    def ImageProcess(self):
+        dicts = {}
+        for root, dirs, files in os.walk(self.img_dir):
+            if len(dirs) == 0:
+                for file in files:
+                    name = file.split('.')[0].split('_')[-1]
+                    if not name in dicts:
+                        dicts[name] = [os.sep.join([root, file])]
+                    else:
+                        dicts[name] = dicts[name] + [os.sep.join([root, file])]
+        return dicts
+        
+    def updateDataCenter(self):
+        for name in os.listdir(cfg.IMG_DIR_TOTAL):
+            img = cv2.imread(os.sep.join([cfg.IMG_DIR_TOTAL, name]))
+            for an in cfg.ANGLES:
+                image = rotate_bound(img, an)
+                cv2.imwrite(os.sep.join([cfg.IMG_DIR_TOTAL, str(an)+'_'+name]), image)
+        self.updateMonitor.update(self.net, self.transform, self.ms)
+        for name in os.listdir(cfg.IMG_DIR_TOTAL):
+            os.remove(os.sep.join([cfg.IMG_DIR_TOTAL, name]))
+
+    def disposaldata(self): 
+        with open(cfg.SAMPLE, 'r', encoding='utf-8') as jsonpath:
+            loadDict = json.load(jsonpath)
+            allsample = set(loadDict['monitor'])
+        for imgname in os.listdir(cfg.TEST_IMG_DIR):
+            imgbarcode = imgname.split('.')[0].split('_')[-1]
+            if not os.path.isdir(cfg.DATA_POOLING):
+                os.mkdir(cfg.DATA_POOLING)
+            if not imgbarcode in allsample:
+#                shutil.move(
+#                    os.sep.join([cfg.TEST_IMG_DIR, imgname]),
+#                    os.sep.join([cfg.DATA_POOLING, imgname]))
+                os.remove(os.sep.join([cfg.TEST_IMG_DIR, imgname]))
+            else:
+                shutil.move(
+                    os.sep.join([cfg.TEST_IMG_DIR, imgname]),
+                    os.sep.join([cfg.SAMPLEDIR, imgname]))
+        for vname in os.listdir(cfg.VIDEOPATH):
+            os.remove(os.sep.join([cfg.VIDEOPATH, vname]))
+
+if __name__ == '__main__':
+    Update = update()
+    Update.disposaldata()
+    #Update.updateDataCenter()