
简介
该用户还未填写简介
擅长的技术栈
未填写擅长的技术栈
可提供的服务
暂无可提供的服务
用Pytorch实现图像分类
概述本文记录使用pytorch深度学习框架来实现一个图像分类任务的过程,搭建一个图像分类的工程主要包括以下过程:1.定义一个加载数据的类2.构建网络结构3.编写训练代码4.编写验证/测试代码一、 定义数据类pytorch中提供了两个类用于训练数据的加载,分别是torch.utils.data.Dataset和 torch.utils.data.DataLoader。Dataset...
机器学习--身份证号识别
import cv2import osimport numpy as npfrom sklearn import svmfrom sklearn import neighborsfrom sklearn.neural_network import MLPClassifier# 垂直投影,分割字符def verticle_projection(thresh1):h, w = thresh1.shap
TensorRT介绍、安装和测试
1. What is TensorRT ? TensorRT的核心是一个C++库,能在NVIDIA的图像处理单元(GPU)上进行高性能的推断。它为与TensorFlow, Caffe, PyTorch, MXNet等主流训练框架进行互补工作而设计,专注于在GPU上高效地运行训练好的网络,生成预测结果。 ..
到底了







