diff --git a/src/prepare/calculate.py b/src/prepare/calculate.py index 7182c26..6ab2cb6 100644 --- a/src/prepare/calculate.py +++ b/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]) diff --git a/src/prepare/spider.py b/src/prepare/spider.py index 0373296..f088975 100644 --- a/src/prepare/spider.py +++ b/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')