七月在线《关键点检测概览与环境配置》

七月在线课程：https://www.julyedu.com/course/getDetail/262

什么是关键点？

关键点定义：关键点也称为兴趣点，它是2D图像、3D点云或曲面模型上，可以通过定义检测标准来获取的具有稳定性、区别性的点集。关键点检测涉及同时检测人和定位他们的关键点。关键点与兴趣点相同。它们是空间位置或图像中的点，它们定义了图像中有趣或突出的内容。它们对图像旋转、收缩、平移、失真等是不变的。

关键点的意义？

加快后续识别、追踪等数据的处理速度。

环境配置

nvidia GPU 配置：

https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html

code ： MNIST

MNIST实战！

import torch
from torchvision import datasets, transforms
import matplotlib.pyplot as plt
import os
import torchvision
import numpy as np
from torch.autograd import Variable
import random
%matplotlib inlinetransform = transforms.Compose([
transforms.ToTensor(), transforms.Normalize([0.5], [0.5])])data_train = datasets.MNIST(root = "./data/",transform=transform,train = True,download = True)data_test = datasets.MNIST(root="./data/",transform = transform,train = False)data_loader_train = torch.utils.data.DataLoader(dataset=data_train,batch_size = 64,shuffle = True,num_workers=2)data_loader_test = torch.utils.data.DataLoader(dataset=data_test,batch_size = 64,shuffle = True,num_workers=2)images, labels = next(iter(data_loader_train))
img = torchvision.utils.make_grid(images)img = img.numpy().transpose(1,2,0)
std = [0.5,0.5,0.5]
mean = [0.5,0.5,0.5]
img = img*std+mean
print([labels[i] for i in range(64)])
plt.imshow(img)class Model(torch.nn.Module):def __init__(self):super(Model, self).__init__()self.conv1 = torch.nn.Sequential(torch.nn.Conv2d(1,64,kernel_size=3,stride=1,padding=1),torch.nn.ReLU(),torch.nn.Conv2d(64,128,kernel_size=3,stride=1,padding=1),torch.nn.ReLU(),torch.nn.MaxPool2d(stride=2,kernel_size=2))self.dense = torch.nn.Sequential(torch.nn.Linear(14*14*128,1024),torch.nn.ReLU(),torch.nn.Dropout(p=0.5),torch.nn.Linear(1024, 10))def forward(self, x):x = self.conv1(x)x = x.view(-1, 14*14*128)x = self.dense(x)return xcost = torch.nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters())
n_epochs = 5for epoch in range(n_epochs):running_loss = 0.0running_correct = 0print("Epoch {}/{}".format(epoch, n_epochs))print("-"*10)for data in data_loader_train:X_train, y_train = dataX_train, y_train = Variable(X_train), Variable(y_train)outputs = model(X_train)_,pred = torch.max(outputs.data, 1)optimizer.zero_grad()loss = cost(outputs, y_train)loss.backward()optimizer.step() #进行单次优化running_loss += loss.datarunning_correct += torch.sum(pred == y_train.data)testing_correct = 0for data in data_loader_test:X_test, y_test = dataX_test, y_test = Variable(X_test), Variable(y_test)outputs = model(X_test)_, pred = torch.max(outputs.data, 1)testing_correct += torch.sum(pred == y_test.data)print("Loss is:{:.4f}, Train Accuracy is:{:.4f}%, Test Accuracy is:{:.4f}".format(running_loss/len(data_train),100*running_correct/len(data_train),100*testing_correct/len(data_test)))
torch.save(model.state_dict(), "model_parameter.pkl")

reference resources

https://paperswithcode.com/sota/keypoint-detection-on-coco-test-dev

七月在线《关键点检测概览与环境配置》相关推荐

mmrotate旋转目标检测框架从环境配置到训练自己的数据集
mmrotate旋转目标检测框架从环境配置到训练自己的数据集最近有旋转目标检测的需求了,在寻找旋转目标检测算法的过程中,发现已经有人提前测评了各类旋转目标检测框架了,结果是mmrorate效果最好. ...
PaddleHub人体骨骼关键点检测(2.0环境)
目录 PaddleHub人体骨骼关键点检测 human_pose_estimation_resnet50_mpii 一.安装新版Hub 二.定义待预测照片三.API预测四.命令行预测五.实现效果 ...
手把手教你用yolov3模型实现目标检测教程(一) - 环境配置
手把手教你用yolov3模型实现目标检测(一) 写在前面: 由于项目需要,使用yolov3模型做了各种现实场景物体的目标检测.做完了过了好长时间,感觉有些遗忘,还是该留下点东西,方便自己查找,也希望能 ...
目标检测FCOS网络环境配置、训练自己的数据集
文章:FCOS: Fully Convolutional One-Stage Object Detection 项目源码:https://github.com/tianzhi0549/FCOS 目录 ...
Mediapipe人脸关键点检测
Mediapipe人脸关键点检测 Mediapipe介绍环境部署人脸关键点检测模型调用与配置输出结果解析绘制面网视频演示 Mediapipe介绍 MediaPipe是由google制作的开 ...
oracle pl sql安装配置,oracle的安装与plsql的环境配置
1,首先得有oracle的安装包和plsql的安装包,安装包地址可见百度云 http://pan.baidu.com/s/1miTqhmg 2.解压下来进入0817账套,找到set.exe文件,双击安 ...
【我的第一个目标检测课题】1、课题介绍及环境配置（Windows+Keras+GPU）
我的第一篇CSDN博客来了. 新手驾到,多多指教. 2020.12.28晚记写在前面今天验收完图像处理课程的结课课题--建筑材料识别课题了.晚上决定趁热打铁,记录一下完成这次课题的全过程,还有一点 ...
深度学习自学（六）：Android人脸检测环境配置等相关问题
Android人脸检测环境配置等相关问题 1.android studio 包查找不到 2.android studio 快捷键搜索类的快捷键是Ctrl+N,搜索以后点击结果,可以跳到相应类的文件. ...
七月在线《图像识别与检测》
七月在线课程:https://www.julyedu.com/course/getDetail/262 CSPNet CSPNet全称是Cross Stage Partial Network,主要从 ...

七月在线《关键点检测概览与环境配置》

什么是关键点？

关键点的意义？

环境配置

code ： MNIST

七月在线《关键点检测概览与环境配置》相关推荐

最新文章

热门文章

七月在线 《关键点检测概览与环境配置》

什么是关键点？

关键点的意义？

环境配置

code ： MNIST

七月在线 《关键点检测概览与环境配置》相关推荐

最新文章

热门文章

七月在线《关键点检测概览与环境配置》

七月在线《关键点检测概览与环境配置》相关推荐