전체 글

경험 정리용 연습장
파이썬

[파이썬] OpenCV, YOLOv4, OpenPose를 이용하여 사진 속 객체 탐지 및 인물의 자세 추정

CUDA와 같이 빌드가 되어있는 4.4.0버전 이상의 OpenCV가 필요함 import cv2 import numpy as np from data.proto_file_path import * from data.weights_file_path import * from data.BODY_PARTS import * from data.POSE_PAIRS import * from data.yolo_classes import classes 필요한 모듈 import data 파일 구성은 아래 포스팅에 나와있음 https://hanryang1125.tistory.com/11 openpose, yolo에 필요한 변수 파일 # 학습시킨 데이터셋에 따른 신체부위 숫자와 이름 BODY_PARTS_MPI = {0: "Head"..

윈도우

[윈도우] NVDIA CUDA 및 cuDNN 설치

1. 본인의 GPU에 맞는 Compute Capability 확인 https://developer.nvidia.com/cuda-gpus CUDA GPUs Recommended GPU for Developers NVIDIA TITAN RTX NVIDIA TITAN RTX is built for data science, AI research, content creation and general GPU development. Built on the Turing architecture, it features 4608, 576 full-speed mixed precision Tensor Cores for accel developer.nvidia.com 2. Compute Capability에 맞는 CUDA SDK..

파이썬

[파이썬] OpenPose, YOLO에 필요한 변수 파일

# 학습시킨 데이터셋에 따른 신체부위 숫자와 이름 BODY_PARTS_MPI = {0: "Head", 1: "Neck", 2: "RShoulder", 3: "RElbow", 4: "RWrist", 5: "LShoulder", 6: "LElbow", 7: "LWrist", 8: "RHip", 9: "RKnee", 10: "RAnkle", 11: "LHip", 12: "LKnee", 13: "LAnkle", 14: "Chest", 15: "Background"} BODY_PARTS_COCO = {0: "Nose", 1: "Neck", 2: "RShoulder", 3: "RElbow", 4: "RWrist", 5: "LShoulder", 6: "LElbow", 7: "LWrist", 8: "RHip", 9: ..

윈도우

[윈도우] CMake를 이용하여 opencv, opencv_contrib, cuda, eigen 빌드

본인의 GPU에 맞게 cuda 및 cudnn은 설치했다고 가정하고 진행 Git을 이용하여 opencv와 opencv_contrib을 다운로드 받고, https://github.com/opencv OpenCV OpenCV has 16 repositories available. Follow their code on GitHub. github.com eigen은 아래 사이트에서 압축파일을 다운로드 받고 압축 해제 http://eigen.tuxfamily.org/index.php?title=Main_Page Eigen Eigen is a C++ template library for linear algebra: matrices, vectors, numerical solvers, and related algorit..

파이썬

[파이썬] OpenCV, YOLO를 이용하여 이미지 속 객체 인식

import cv2 import numpy as np 필요한 모듈 import def yolo(frame, size, score_threshold, nms_threshold): # YOLO 네트워크 불러오기 net = cv2.dnn.readNet(f"yolov3_{size}.weights", "yolov3.cfg") layer_names = net.getLayerNames() output_layers = [layer_names[i[0] - 1] for i in net.getUnconnectedOutLayers()] # 클래스의 갯수만큼 랜덤 RGB 배열을 생성 colors = np.random.uniform(0, 255, size=(len(classes), 3)) # 이미지의 높이, 너비, 채널 받아오기..

IntegerString
코딩 연습장