logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

用Pytorch实现图像分类

概述本文记录使用pytorch深度学习框架来实现一个图像分类任务的过程,搭建一个图像分类的工程主要包括以下过程:1.定义一个加载数据的类2.构建网络结构3.编写训练代码4.编写验证/测试代码一、 定义数据类pytorch中提供了两个类用于训练数据的加载,分别是torch.utils.data.Dataset和 torch.utils.data.DataLoader。Dataset...

#深度学习#pytorch#神经网络
机器学习--身份证号识别

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上高效地运行训练好的网络,生成预测结果。  ..

#深度学习#caffe#tensorflow +1
到底了