更新同步master

src/prepare/calculate.py

@@ -4,9 +4,41 @@
 # @Software: PyCharm
 # @Function:
 import os
+import sys
+import subprocess
+import json
+import re
+sys.path.append(os.path.dirname(os.path.abspath(__file__)) + "/../../PaddleClas-release-2.3")
+from Search_2D import Astar, plotting, env
 
-os.chdir(r'E:\PaddleClas-release-2.3\PaddleClas-release-2.3\deploy')
 
+def main():
+    s_start = (5, 5)
+    s_goal = (45, 25)
+
+    astar = Astar.AStar(s_start, s_goal, "euclidean")
+    plot = plotting.Plotting(s_start, s_goal)
+
+    path, visited = astar.searching()
+    plot.animation(path, visited, "A*")  # animation
+
+# if __name__ == '__main__':
+#     main()
+
+os.chdir(r'E:\PaddleClas-release-2.3\PaddleClas-release-2.3\deploy')
 ####
 
-os.system('python python/predict_system.py -c configs/inference_general.yaml -o Global.infer_imgs="./Trees/test_images/90.jpeg" -o IndexProcess.index_dir="./Trees/index" -o Global.use_gpu=False -o IndexProcess.Image_root="./Trees/gallery/" -o IndexProcess.data_file="./Trees/gallery/tree_list.txt')
+result = subprocess.Popen('python python/predict_system.py -c configs/inference_general.yaml -o Global.infer_imgs="./Trees/test_images/90.jpeg" -o IndexProcess.index_dir="./Trees/index" -o Global.use_gpu=False -o IndexProcess.Image_root="./Trees/gallery/" -o IndexProcess.data_file="./Trees/gallery/tree_list.txt',
+                          shell=False,
+                          stdout=subprocess.PIPE,
+                          stderr=subprocess.STDOUT,
+                          encoding='utf-8',
+                          )
+
+string = result.communicate()[0].split("per batch image")[1].replace("'", '"').strip()
+ansi_escape = re.compile(r'\n\x1b(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])')
+result = ansi_escape.sub('', string)
+data_json = json.loads(result)
+print(data_json[0])
+print(data_json[1])
+print(data_json[2])

src/prepare/spider.py

@@ -3,3 +3,81 @@
 # @File : spider.py 
 # @Software: PyCharm
 # @Function:
+import requests
+from os import makedirs
+from os.path import exists
+import logging
+
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s: %(message)s')
+
+header = {
+    'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 11_1_0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36'
+}
+
+RESULTS_DIR = 'tupian'  # 结果保存路径
+exists(RESULTS_DIR) or makedirs(RESULTS_DIR)
+
+n = 0
+pn = 1
+# pn是从第几张图片获取 百度图片下滑时默认一次性显示30张
+
+page = 20
+page = int(page) + 1
+
+for m in range(1, page):
+    url = 'https://image.baidu.com/search/acjson?'
+    param = {
+        'tn': 'resultjson_com',
+        'logid': '8846269338939606587',
+        'ipn': 'rj',
+        'ct': '201326592',
+        'is': '',
+        'fp': 'result',
+        'queryWord': '车牌号',
+        'cl': '2',
+        'lm': '-1',
+        'ie': 'utf-8',
+        'oe': 'utf-8',
+        'adpicid': '',
+        'st': '-1',
+        'z': '',
+        'ic': '',
+        'hd': '',
+        'latest': '',
+        'copyright': '',
+        'word': '树',
+        's': '',
+        'se': '',
+        'tab': '',
+        'width': '',
+        'height': '',
+        'face': '0',
+        'istype': '2',
+        'qc': '',
+        'nc': '1',
+        'fr': '',
+        'expermode': '',
+        'force': '',
+        'cg': 'girl',
+        'pn': pn,  # 从第几张图片开始
+        'rn': '30',
+        'gsm': '1e',
+    }
+    page_text = requests.get(url=url, headers=header, params=param)
+    page_text.encoding = 'utf-8'
+    page_text = page_text.json()
+    info_list = page_text['data']
+    del info_list[-1]
+    img_path_list = []
+
+    for i in info_list:
+        img_path_list.append(i['thumbURL'])
+    for img_path in img_path_list:
+        img_data = requests.get(url=img_path, headers=header).content
+        img_path = f'./{RESULTS_DIR}/' + str(n) + '.jpg'
+        with open(img_path, 'wb') as fp:
+            fp.write(img_data)
+        n = n + 1
+        logging.info(f'download page{m} photoes {n} successed')
+    pn += 29
+    logging.info(f'download page{m} successed')