[UPDATE] 기존 코드 수정

 
 
 import time
 import time
+import math
 import cv2
 import cv2
 import numpy as np
 import numpy as np
 from model.yolo_model import YOLO
 from model.yolo_model import YOLO
 
 
+class Point2D:
+    def __init__(self, width, height):
+        self.width = width
+        self.height = height
 
 
 def process_image(img):
 def process_image(img):
     image = cv2.resize(img, (416, 416),
     image = cv2.resize(img, (416, 416),
@@ -53,7 +58,11 @@ if __name__ == '__main__':
         name = all_classes[cl]
         name = all_classes[cl]
         size = w*h
         size = w*h
         # 얼마나 가운데인지 확인하는 알고리즘
         # 얼마나 가운데인지 확인하는 알고리즘
-        coordinatevalue = abs((x+0.5*w)-image.shape[1]/2) + abs((y-h*0.5)-image.shape[0]/2)/(image.shape[0]/image.shape[1])
+        
+        object = Point2D(width= x + 0.5 * w  , height= y - h * 0.5)
+        a = image.shape[1]/2 - object.width
+        b = image.shape[0]/2 - object.height
+        coordinatevalue = math.sqrt((a*a)+(b*b))
         detectionInfo.append([i, name, size, coordinatevalue])
         detectionInfo.append([i, name, size, coordinatevalue])
 
 
 # name 별로 크기가 가장 크거나 물체가 프레임의 가운데 있는 프레임 번호 목록 얻어오기
 # name 별로 크기가 가장 크거나 물체가 프레임의 가운데 있는 프레임 번호 목록 얻어오기