# Train_Custom_Dataset

**Repository Path**: simo2018/Train_Custom_Dataset

## Basic Information

- **Project Name**: Train_Custom_Dataset
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2024-12-15
- **Last Updated**: 2024-12-15

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

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

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

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

代码测试[云GPU环境](https://featurize.cn?s=d7ce99f842414bfcaea5662a97581bd1)：GPU RTX 3060、CUDA v11.2

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

![计算机视觉解决的基本问题](https://zihao-download.obs.cn-east-3.myhuaweicloud.com/img_bed/20220803/cv_fund.png)

[优雅地感谢子豪兄](https://github.com/TommyZihao/zihao_commercial)

## 图像分类

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

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

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://zdb.h5.xeknow.com/s/3MPxbI)

## 单目标追踪（蜜蜂追踪）

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