4.6(Tues)_project

센터환경

num_evalnew_test.py를 돌려보았다.(eval(revise).py에서 원하는 객체만 잡히게 code추가한거)

drive.google.com/drive/folders/1nOS4LhRMaW9o_SFPWOc43iY8yeaWQZIW

Google Drive - 모든 파일을 한 곳에서

 

**참조 github

github.com/dbolya/yolact/blob/master/eval.py

dbolya/yolact

 

num_evalnew_test.py파일(eval.py=>code추가)에서

 

t = postprocess(dets_out, w, h, visualize_lincomb = args.display_lincomb,

crop_masks = args.crop,

score_threshold = args.score_threshold)

cfg.rescore_bbox = save

#print(lenth(t))

print(t[3][0][0])

##return값이 class, score, box, mask순으로 되있음

여기서 허덕였다...

 

 

 

layers folder에 output_utils.py파일 부분에서 

 

for jdx in range(masks.size(0)):

x1, y1, x2, y2 = boxes[jdx, :]

mask_w = x2 - x1

mask_h = y2 - y1

# Just in case

if mask_w * mask_h <= 0 or mask_w < 0:

continue mask = masks[jdx, :].view(1, 1, cfg.mask_size, cfg.mask_size)

mask = F.interpolate(mask, (mask_h, mask_w), mode=interpolation_mode, align_corners=False)

mask = mask.gt(0.5).float()

full_masks[jdx, y1:y2, x1:x2] = mask

masks = full_masks

return classes, scores, boxes, masks

 

이 부분 하나하나 뜯어보면서 객체의 중심값(ex_hip)을 기준으로 하여 객체 segmentation반영,

bed값 위에 있는 사람의 중심점을 기준으로 무릎, 어깨가 벗어나면 알람이 울리는 시스템 반영

 

환경 : Galaxy ion2 notebook(CPU : intel-i5, GPU : NVIDIA-mx450)

Visual Studio 2019에서 CUDA를 적용한 OpenCV 빌드해봄

 

CUDA설치, cuDNN설치, Visual Studio에서 openCV build완료

 

・CUDA: CUDA Toolkit 10.1 update2

・cuDNN: 7.6.5

・PyTorch: 1.4.0

・torchvision: 0.5.0

・cudatoolkit: 10.1.243

 

 

 

pip install tensorflow-gpu==2.3.0 설치하고

 

from tensorflow.python.client

import device_lib

device_lib.list_local_devices()

텐서플로우 gpu잡히는지 확인

 

visual studio code에서 

import cv2

print(cv2.__version__)

print(cv2.cuda.getCudaEnabledDeviceCount()) 실행시켜

 

cv2 version이 4.3.0,

1이 잡히는지 확인했다.

이제다시 yolact등 model돌려보자

 

 

 

그럼 이만.

총총.