from flask_restx import Resource, Api, Namespace, fields,reqparse from fog_model.fog_predict import fog_classfication from fog_model.fog_model import darkchannel from flask import request from flask import Flask, render_template, request from werkzeug.utils import secure_filename import os from haversine import haversine from database.database import DB import torch from PIL import Image from pothole_model.pothole import pothole paths = os.getcwd() Action = Namespace( name="Action", description="노드 분석을 위해 사용하는 api.", ) def find_node(gps_address_y,gps_address_x): db=DB() nn_end = None end_delta = float("inf") value=0.0001 near_nodes=[] while near_nodes==[]: value=value*10 near_nodes=db.db_get_near_node(gps_address_y,gps_address_x,value) for n in near_nodes: e_dist = haversine((gps_address_x,gps_address_y), n) if e_dist < end_delta : nn_end = n end_delta = e_dist return nn_end @Action.route('/image_summit') class fileUpload(Resource): @Action.doc(responses={200: 'Success'}) @Action.doc(responses={500: 'Register Failed'}) def post(self): if request.method == 'POST': f = request.files['file'] f.save(secure_filename(f.filename)) return { 'save': 'done' # str으로 반환하여 return }, 200 @Action.route('/image_anal') class fileUpload(Resource): @Action.doc(responses={200: 'Success'}) @Action.doc(responses={500: 'Register Failed'}) def post(self): if request.method == 'POST': fc = fog_classfication() dir = os.getcwd() filename = request.json['filename'] file_type = request.json['file_type'] gps_address_x = float(request.json['gps_x']) gps_address_y = float(request.json['gps_y']) total_path = dir+ "\\"+ filename + file_type model_fc= fc.predict(total_path) model_yolo= torch.hub.load(paths +'/yolov5/', 'custom', path=paths+'/yolov5/best.pt', source='local') im = Image.open(total_path) results = model_yolo(im) li_detect=list(results.pandas().xyxy[0]['name']) if model_fc == "normal": if 'vest' in li_detect and 'cone' in li_detect: nn_end = find_node(gps_address_y,gps_address_x) return { 'node': nn_end, 'fog' : 'normal', 'construction' : 'construction' }, 200 else: return { 'node': None, 'fog' : 'normal', 'construction' : 'normal' }, 200 else: if 'vest' in li_detect and 'cone' in li_detect: nn_end = find_node(gps_address_y,gps_address_x) return { 'node': nn_end, 'fog' : 'fog', 'construction' : 'construction' }, 200 else: nn_end = find_node(gps_address_y,gps_address_x) return { 'node': nn_end, 'fog' : 'fog', 'construction' : 'normal' }, 200 @Action.route('/pothole_report') class fileUpload(Resource): @Action.doc(responses={200: 'Success'}) @Action.doc(responses={500: 'Register Failed'}) def post(self): if request.method == 'POST': pc = pothole() dir = os.getcwd() report_id = request.json['report_id'] pothole_id = request.json['pothole_id'] pothole_x = float(request.json['pothole_x']) pothole_y = float(request.json['pothole_y']) z = float(request.json['z']) pc.report(report_id,pothole_id,pothole_x,pothole_y,z) return { 'report': 'done' # str으로 반환하여 return }, 200 @Action.route('/pothole_display') class fileUpload(Resource): @Action.doc(responses={200: 'Success'}) @Action.doc(responses={500: 'Register Failed'}) def post(self): if request.method == 'POST': pc = pothole() dir = os.getcwd() timestamp = request.json['timestamp'] value=pc.display(timestamp) return { 'pothole': list(value) # str으로 반환하여 return }, 200