Giter Club home page Giter Club logo

train_custom_dataset's Introduction

子豪兄带你两天搞定AI毕业设计

标注自己的数据集,训练、评估、测试、部署自己的人工智能算法

作者:同济子豪兄 https://space.bilibili.com/1900783

代码测试云GPU环境:GPU RTX 3060、CUDA v11.2

本教程的数据集、代码、视频,倾注了子豪兄大量时间和心血。如果知识付费,卖两三千并不为过,但本着开源分享精神,全部免费开源,但仅可用于教学、科研、科普等非盈利用途,并需在转载引用时注明出处。

计算机视觉解决的基本问题

优雅地感谢子豪兄

图像分类

构建自己的图像分类数据集

收集图像、下载样例数据集,删除系统多余文件,划分训练集、测试集,统计图像尺寸、比例分布、拍照地点位置分布,统计各类别图像数量

https://www.bilibili.com/video/BV1Jd4y1T7rw

【Pytorch】ImageNet预训练图像分类模型预测

使用Pytorch自带的预训练图像分类模型,分别对单张图像、视频、摄像头实时画面运行图像分类预测

https://www.bilibili.com/video/BV1qe4y1D7zD

【Pytorch】迁移学习Fine-tuning训练自己的图像分类模型

https://www.bilibili.com/video/BV1Ng411C7WY

【Pytorch】用训练得到的pytorch图像分类模型,识别图像、视频、摄像头画面

https://www.bilibili.com/video/BV12d4y1P7xz

测试集评估

计算各类别分类评估指标,绘制混淆矩阵、PR曲线、ROC曲线。

https://www.bilibili.com/video/BV1bP411j7NK

测试集语义特征降维可视化

抽取Pytorch训练得到的图像分类模型中间层的输出特征,作为输入图像的语义特征。计算测试集所有图像的语义特征,使用t-SNE和UMAP两种降维方法降维至二维和三维,可视化。

https://www.bilibili.com/video/BV1VB4y1z7xN

可解释性分析、显著性分析

CAM热力图系列算法:https://www.bilibili.com/video/BV1JG4y1s74x

答疑交流群

子豪兄图像分类答疑交流群(有问必答)

单目标追踪(蜜蜂追踪)

https://www.bilibili.com/video/BV1za411Y7Zm

视频人流量计数+足迹追踪

https://www.bilibili.com/video/BV1za411Y7Zm

大模型摘要生成

https://www.bilibili.com/video/BV1W44y1g7cB

CycleGAN图像风格迁移

https://www.bilibili.com/video/BV1wv4y1T71F

OCR文字识别

https://www.bilibili.com/video/BV1Ua411x7dB

train_custom_dataset's People

Contributors

tommyzihao avatar ethan-chen-plus avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.