Python

코랩

[코랩] 영상 속 객체 탐지 및 인물의 자세 추정 (OpenCV, YOLOv4, OpenPose)

import cv2 import numpy as np from google.colab.patches import cv2_imshow from tqdm import tqdm protoFile_body_25b = "pose_deploy.prototxt" weightsFile_body_25b = "pose_iter_XXXXXX.caffemodel" BODY_PARTS_BODY_25B = {0: "Nose", 1: "LEye", 2: "REye", 3: "LEar", 4: "REar", 5: "LShoulder", 6: "RSoulder", 7: "LElbow", 8: "RElbow", 9: "LWrist", 10: "RWrist", 11: "LHip", 12: "RHip", 13: "LKnee", 14: "R..

파이썬

[파이썬] OpenCV, OpenPose를 이용하여 사진 속 인물의 자세 추정 - Experimental Model 추가

https://github.com/CMU-Perceptual-Computing-Lab/openpose_train/tree/master/experimental_models#body-25b-model-option-2-recommended CMU-Perceptual-Computing-Lab/openpose_train Training repository for OpenPose. Contribute to CMU-Perceptual-Computing-Lab/openpose_train development by creating an account on GitHub. github.com pose_iter_XXXXXX.caffemodel 다운로드 (가중치 파일) https://github.com/CMU-Perceptua..

파이썬

[파이썬] 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)) # 이미지의 높이, 너비, 채널 받아오기..

파이썬

다리를 지나는 트럭 (Python Queue)

def solution(bridge_length, weight, truck_weights): bridge_list = [] sum = 0 count = 0 while True: if count > 0 and sum == 0: break if len(bridge_list) > bridge_length - 1: sum -= bridge_list[0] bridge_list.pop(0) if len(truck_weights) != 0: if sum + truck_weights[0]

파이썬

OpenCV, OpenPose, math를 이용하여 옆모습 사진 속 인물의 허리가 숙여졌는지 추정 (Python)

import cv2 import math 필요한 모듈 import def output_keypoints(image_path, proto_file, weights_file, threshold, model_name, BODY_PARTS): global points # 이미지 읽어오기 frame = cv2.imread(image_path) # 네트워크 불러오기 net = cv2.dnn.readNetFromCaffe(proto_file, weights_file) # 입력 이미지의 사이즈 정의 image_height = 368 image_width = 368 # 네트워크에 넣기 위한 전처리 input_blob = cv2.dnn.blobFromImage(frame, 1.0 / 255, (image_width, im..

IntegerString
'Python' 태그의 글 목록 (3 Page)