Page 502 - 2020학년도 MDP과제발표회 자료집 (통신과) (3)

orig_im = img
dim = orig_im.shape[1], orig_im.shape[0]
img = cv2.resize(orig_im, (inp_dim, inp_dim))
img_ = img[:, :, ::-1].transpose((2, 0, 1)).copy()
img_ = torch.from_numpy(img_).float().div(255.0).unsqueeze(0)
return img_, orig_im, dim

def write(self, x, img):

c1 = tuple(x[1:3].int())
c2 = tuple(x[3:5].int())
cls = int(x[-1])
label = "{0}".format(str(self.classes[cls])) #str 을 통해 변수 크기 확장
self.label_list.append(label)
color = random.choice(self.colors)
cv2.rectangle(img, c1, c2, color, 1)
t_size = cv2.getTextSize(label, cv2.FONT_HERSHEY_PLAIN, 1, 1)[0]
c2 = c1[0] + t_size[0] + 3, c1[1] + t_size[1] + 4
cv2.rectangle(img, c1, c2, color, -1)
cv2.putText(img, label, (c1[0], c1[1] + t_size[1] + 4), cv2.FONT_HERSHEY_PLAIN, 1,
[225, 255, 255], 1);
return img

def arg_parse(self):
"""
Parse arguements to the detect module
"""
parser = argparse.ArgumentParser(description='YOLO v3')
parser.add_argument("--confidence", dest="confidence", help="Object Confidence to
filter predictions", default=0.25)
parser.add_argument("--nms_thresh", dest="nms_thresh", help="NMS Threshhold",
default=0.4)
parser.add_argument("--reso", dest='reso', help=
"Input resolution of the network. Increase to increase accuracy. Decrease to increase
speed",
default="160", type=str)
return parser.parse_args()

524 인천전자마이스터고등학교 - 524 -
정보통신기기과