강세희

[UPDATE] 기존 코드 수정

import time
import math
import cv2
import numpy as np
from model.yolo_model import YOLO
class Point2D:
def __init__(self, width, height):
self.width = width
self.height = height
def process_image(img):
image = cv2.resize(img, (416, 416),
......@@ -53,7 +58,11 @@ if __name__ == '__main__':
name = all_classes[cl]
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])
# name 별로 크기가 가장 크거나 물체가 프레임의 가운데 있는 프레임 번호 목록 얻어오기
......