def check_dataset(data, autodownload=True): """用在train.py和detect.py中 检查本地有没有数据集 检查数据集 如果本地没有则从torch库中下载并解压数据集 :params data: 是一个解析过的data_dict len=7 例如: ['path'='../datasets/coco128', 'train','val', 'test', 'nc', 'names', 'downl...
1、使用darknet训练处的YOLOv3权重文件的大小一般在200M+左右,这对于在应用阶段做模型加载的时候时间非常的长,因此我们项目组打算对训练好的模型做裁剪,就是把一些不必要的卷积核及其参数删除掉,(删除的原则就是设置一阈值,当小于某一个阈值的时候踢出该卷积核和对应的参数),在此基础对裁剪完成后的权重文件重新fin...