jetson nano 人体姿态识别

源码在git

jetpack 版本　4.2 ,直接安装 gluoncv ，如果需要能直接跑nano的镜像，可以加入星球，

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Thu Jul  1 17:17:29 2021
@author: ledi
"""import cv2def gstreamer_pipeline(capture_width=1280,capture_height=720,display_width=1280,display_height=720,framerate=60,flip_method=0,
):return ("nvarguscamerasrc ! ""video/x-raw(memory:NVMM), ""width=(int)%d, height=(int)%d, ""format=(string)NV12, framerate=(fraction)%d/1 ! ""nvvidconv flip-method=%d ! ""video/x-raw, width=(int)%d, height=(int)%d, format=(string)BGRx ! ""videoconvert ! ""video/x-raw, format=(string)BGR ! appsink"% (capture_width,capture_height,framerate,flip_method,display_width,display_height,))# def show_camera():
#     cap = cv2.VideoCapture(gstreamer_pipeline(flip_method=0), cv2.CAP_GSTREAMER)#     while cap.isOpened():
#         flag, img = cap.read()
#         cv2.imshow("CSI Camera", img)
#         kk = cv2.waitKey(1)#         # do other things#         if kk == ord('q'):  # 按下 q 键，退出
#             break#     cap.release()
#     cv2.destroyAllWindows()#from __future__ import division
import argparse, time, logging, os, math, tqdm, cv2import numpy as np
import mxnet as mx
from mxnet import gluon, nd, image
from mxnet.gluon.data.vision import transformsimport matplotlib.pyplot as pltimport gluoncv as gcv
from gluoncv import data
from gluoncv.data import mscoco
from gluoncv.model_zoo import get_model
from gluoncv.data.transforms.pose import detector_to_simple_pose, heatmap_to_coord
from gluoncv.utils.viz import cv_plot_image, cv_plot_keypointsctx = mx.cpu()
detector_name = "ssd_512_mobilenet1.0_coco"
detector = get_model(detector_name, pretrained=True, ctx=ctx)detector.reset_class(classes=['person'], reuse_weights={'person':'person'})
detector.hybridize()estimator = get_model('simple_pose_resnet18_v1b', pretrained='ccd24037', ctx=ctx)
estimator.hybridize()#cap = cv2.VideoCapture(0)
time.sleep(1)  ### letting the camera autofocusaxes = None
num_frames = 100cap = cv2.VideoCapture(gstreamer_pipeline(flip_method=0), cv2.CAP_GSTREAMER)while cap.isOpened():flag, frame = cap.read()print(11111)# cv2.imshow("CSI Camera", img)# kk = cv2.waitKey(1)# # do other things# if kk == ord('q'):  # 按下 q 键，退出#     break# ret, frame = cap.read()frame = mx.nd.array(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)).astype('uint8')print(2222)x, frame = gcv.data.transforms.presets.ssd.transform_test(frame, short=512, max_size=350)print(333)x = x.as_in_context(ctx)class_IDs, scores, bounding_boxs = detector(x)print(444)pose_input, upscale_bbox = detector_to_simple_pose(frame, class_IDs, scores, bounding_boxs,output_shape=(128, 96), ctx=ctx)print(pose_input)if len(upscale_bbox) > 0:predicted_heatmap = estimator(pose_input)pred_coords, confidence = heatmap_to_coord(predicted_heatmap, upscale_bbox)img = cv_plot_keypoints(frame, pred_coords, confidence, class_IDs, bounding_boxs, scores,box_thresh=0.5, keypoint_thresh=0.2)cv_plot_image(img)if cv2.waitKey(1) & 0xFF == ord('q'):# 存储图片cv2.imwrite("camera.jpeg", frame)break#python cam_demo.py --num-frames 100

jetson nano 人体姿态识别相关推荐

agx 安装ros opencv_【ROS开源】Jetson Nano阿克曼转向机器人
塔克创新出品的系列入门ROS智能机器人学习开发平台,包括麦克纳姆轮机器人,四轮差速机器人,三轮全向机器人,两轮差速机器人,阿克曼转向小车等平台.可实现SLAM建图导航.自动避障.雷达跟随.视觉巡线.三 ...
Jetson nano 树莓派传输视频帧直播极低延迟
最近要实现一个项目,就是使用Jetson nano,进行人体识别,同时推流视频到公网. 最初的版本是,在Nano跑YOLOV5,再将处理后的图片经过ffmpeg推送到自己搭建的nginx直播服务器,服 ...
jetson nano 用 tensorrt 运行 nanodet(kitti数据集)
题目是目标,我们先一步一步来做,第一步是训练神经网络, 我用的是 kitti 数据集,训练顺序为,第一步,拿到kitti数据集,第二步,把kitti数据集修改为voc数据集格式,第三步,修改配置文件进 ...
树莓派4与英伟达Jetson Nano性能大比拼，谁是最佳的嵌入式“电脑”？
作者 | Chris Pietschmann 译者 | 弯月,责编 | 屠敏转载自CSDN(ID:CSDNnews) 导读:日前,Raspberry 基金会发布了开发者为之兴奋的 Raspberry ...
jetson nano 人脸
jetson nano 人脸 https://github.com/SteveMacenski/jetson_nano_detection_and_tracking/blob/6420430868aa ...
Jetson nano安装JupyterLab
Jetson nano安装JupyterLab 1 先安装JupyterLab依赖 1.1 安装JupyterLab依赖及JupyterLab 1.2 安装jupyterlab中可能报错:`c/_cf ...
linux或Jetson nano上安装torchvision或安装pillow报错：The headers or library files could not be found for jpeg
1 错误分析我在Jetson nano上安装torchvision的时候报错(torchvision依赖pillow库):The headers or library files could not ...
在Ubuntu上通过VNC Viewer连接Jetson nano桌面
1 在Ubuntu上安装VNC Viewer客户端 1.首先在自己的电脑Ubuntu系统上安装VNC Viewer客户端 2.下载Linux版本的VNC Viewer客户端软件 VNC Viewer客 ...
Jetson Nano and VIM3硬件参数对比及目标检测性能对比
文章目录: 1 Jetson Nano and VIM3硬件参数对比及目标检测性能对比 2 Jetson nano在yolov4目标检测性能 3 VIM在yolov3.yolov3-tiny.yolo ...
解决Jetson nano上编译librealsense错误：Could NOT find Vulkan (missing: VULKAN_LIBRARY VULKAN_INCLUDE_DIR)
1 我的Jetson nano环境 zhihui@zhihui-desktop:~$ jetson_release -v- NVIDIA Jetson Nano (Developer Kit Vers ...

jetson nano 人体姿态识别

jetpack 版本　4.2 ,直接安装 gluoncv ，如果需要能直接跑nano的镜像，可以加入星球，

jetson nano 人体姿态识别相关推荐

最新文章

热门文章

jetson nano 人体姿态识别

jetpack 版本 4.2 ,直接安装 gluoncv ，如果需要能直接跑nano的镜像，可以加入星球，

jetson nano 人体姿态识别相关推荐

最新文章

热门文章

jetpack 版本　4.2 ,直接安装 gluoncv ，如果需要能直接跑nano的镜像，可以加入星球，