基于CenterNet简化后只做目标检测的源代码
基于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一张图片推
文件列表
基于CenterNet简化后只做目标检测的源代码
(预估有个778文件)
vision.cpp
405B
dcn_v2_im2col_cpu.h
5KB
dcn_v2_psroi_pooling_cpu.cpp
17KB
dcn_v2_cpu.cpp
11KB
dcn_v2_im2col_cpu.cpp
19KB
vision.h
3KB
dcn_v2.h
8KB
dcn_v2_im2col_cuda.cu
20KB
dcn_v2_cuda.cu
14KB
dcn_v2_psroi_pooling_cuda.cu
16KB
暂无评论