1 Star 0 Fork 3

董龙云 / Train_Custom_Dataset

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

子豪兄带你两天搞定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

空文件

简介

同济大学子豪up学习资料 展开 收起
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/donglong-cloud/Train_Custom_Dataset.git
git@gitee.com:donglong-cloud/Train_Custom_Dataset.git
donglong-cloud
Train_Custom_Dataset
Train_Custom_Dataset
main

搜索帮助

344bd9b3 5694891 D2dac590 5694891