Popular repositories Loading

1. TimeSeriesForecasting TimeSeriesForecasting Public

   基于pytorch实现的时间序列预测训练框架，各个部分模块化，方便修改模型。包含时间序列预测模型、训练、验证、测试、wandb可视化、onnx导出、onnx推理、tensorrt导出、tensorrt推理。

   Python 38 4

2. classification classification Public

   基于pytorch实现的图片分类模型训练框架，各个部分模块化，方便修改模型。包含分类模型、训练、验证、测试、剪枝再训练、wandb可视化、onnx导出、onnx推理、tensorrt导出、tensorrt推理、部署。

   Python 16 2

3. insightface insightface Public

   快速建立人脸特征数据库并进行人脸检测和身份识别。

   Python 8 1

4. FaceRecognition FaceRecognition Public

   遮挡人脸识别，在insightface基础上增加了一个识别遮挡概率的分类模型，包含在项目中。

   Python 8 1

5. ObjectDetection ObjectDetection Public

   基于pytorch实现的目标检测训练框架，各个部分模块化，方便修改模型。包含目标检测模型、训练、验证、测试、剪枝再训练、wandb可视化、onnx导出。

   Python 7 1

6. notebook notebook Public

   学习笔记。包含python环境、python基础库、深度学习相关库、深度学习相关理论、其他常用知识。目前文档总页数有130+。

   7 1