Pinned Repositories
TimeSeriesForecasting
基于pytorch实现的时间序列预测训练框架,各个部分模块化,方便修改模型。包含时间序列预测模型、训练、验证、测试、wandb可视化、onnx导出、onnx推理、tensorrt导出、tensorrt推理。
classification
基于pytorch实现的图片分类模型训练框架,各个部分模块化,方便修改模型。包含分类模型、训练、验证、测试、剪枝再训练、wandb可视化、onnx导出、onnx推理、tensorrt导出、tensorrt推理、部署。
FaceRecognition
遮挡人脸识别,在insightface基础上增加了一个识别遮挡概率的分类模型,包含在项目中。
insightface
快速建立人脸特征数据库并进行人脸检测和身份识别。
notebook
学习笔记。包含python环境、python基础库、深度学习相关库、深度学习相关理论、其他常用知识。目前文档总页数有130+。
ObjectDetection
基于pytorch实现的目标检测训练框架,各个部分模块化,方便修改模型。包含目标检测模型、训练、验证、测试、剪枝再训练、wandb可视化、onnx导出。
pphuman
快速使用行人检测模型。基于paddle官方pphuman项目整理,方便快速使用。
clip
快速建立图片特征数据库并用文本描述进行搜索。比如输入'找工作',可以搜到数据库中和'找工作'这个描述相关的图片。支持模型微调训练。
GPT
基于pytorch实现的语言生成模型提示微调框架,各个部分模块化,方便修改模型。包含peft模型训练、知识库检索、大模型与知识库结合、可视化。
ppocr
快速使用OCR模型识别图片中的文字。基于paddle官方ocr项目整理,方便快速使用。
TWK2022's Repositories
TWK2022/TimeSeriesForecasting
基于pytorch实现的时间序列预测训练框架,各个部分模块化,方便修改模型。包含时间序列预测模型、训练、验证、测试、wandb可视化、onnx导出、onnx推理、tensorrt导出、tensorrt推理。
TWK2022/notebook
学习笔记。包含python环境、python基础库、深度学习相关库、深度学习相关理论、其他常用知识。目前文档总页数有130+。
TWK2022/GPT
基于pytorch实现的语言生成模型提示微调框架,各个部分模块化,方便修改模型。包含peft模型训练、知识库检索、大模型与知识库结合、可视化。
TWK2022/ObjectDetection
基于pytorch实现的目标检测训练框架,各个部分模块化,方便修改模型。包含目标检测模型、训练、验证、测试、剪枝再训练、wandb可视化、onnx导出。
TWK2022/classification
基于pytorch实现的图片分类模型训练框架,各个部分模块化,方便修改模型。包含分类模型、训练、验证、测试、剪枝再训练、wandb可视化、onnx导出、onnx推理、tensorrt导出、tensorrt推理、部署。
TWK2022/GNN
基于pytorch实现的图神经网络训练框架,各个部分模块化,方便修改模型。包含图神经网络模型、训练、验证、测试、wandb可视化。
TWK2022/clip
快速建立图片特征数据库并用文本描述进行搜索。比如输入'找工作',可以搜到数据库中和'找工作'这个描述相关的图片。支持模型微调训练。
TWK2022/pphuman
快速使用行人检测模型。基于paddle官方pphuman项目整理,方便快速使用。
TWK2022/ppocr
快速使用OCR模型识别图片中的文字。基于paddle官方ocr项目整理,方便快速使用。
TWK2022/FaceRecognition
遮挡人脸识别,在insightface基础上增加了一个识别遮挡概率的分类模型,包含在项目中。
TWK2022/insightface
快速建立人脸特征数据库并进行人脸检测和身份识别。