1. 首页
  2. 大数据
  3. 算法与数据结构
  4. 基于CenterNet简化后只做目标检测的源代码

基于CenterNet简化后只做目标检测的源代码

上传者: 2023-01-02 09:47:01上传 ZIP文件 201.9MB 热度 7次
基于CenterNet简化后只做目标检测的源代码各个文件说明一目录层级data存放训练数据目前只存放了太平洋汽车网的训练数据主要是识别车牌的位置.只有训练集的数据.exp训练中日志存放的位置images验证时候图片存放的位置models存放训练过程中产生的模型src源代码二文件说明dataset CtdetDataset.py读取图片并进行预处理detector CtdetDetector.py预测推理流程model存放各个BackBone的模型代码prune prune.pySlimming裁剪方法train CtdetTrainer.py训练主体流程CtdetTrainer NNI.py基于NNI裁剪训练主体流程losses.py训练中各种loss的计算方法utils image.py图像处理方法包括高斯变化调整亮度饱和度等方法logger.py日志处理方法src demo3dcar ai.py一张图片推
用户评论