LeeNet分类网络(ImageNet100(已开源):top1-73.44%,top5-91.40%,CPU_forward-30ms,model_size-2.7M)
一、前言
分类网络应用广泛,亦可作为预训练模型用于检测网络,有很好的科研和商业价值。一般的分类网络基于多种数据集做了一个综合的测评,但是因工程条件限制,很难将所有数据集下载并测试,所以博主整理了imagenet100数据集,是从imagenet1000中扣取出来的100类,每类训练数据1000张,测试数据300张,满足工业应用。
二、开源数据集链接:https://pan.baidu.com/s/1PwQFY7ZSnZd5REg38IFaJQ (yxzb )
三、网络
网络改自ResNet,训练出的网络模型大小为2.7MB。
四、测试条件
测试硬件:CPU-i7-4790-8core
测试软件:vs2013,caffe-windows,以及各种依赖库
无、测试结果
测试结果:准确率方面(top1-73.44%,top5-91.40%)
速度方面(forward_CPU_time-30ms)
模型大小(2.7MB)
六、对比结果:
已知各网络对比结果如下(取自YOLO官网),可见博主网络在工业应用中有一定的优势:

七、后续计划:1)下载多个数据集对比,考虑撰写论文
2)考虑公开网络和模型,上传至github,以供科研交流
任何问题请加唯一QQ2258205918(名称samylee) !!!
或唯一VX:samylee_csdn
AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。
更多推荐



所有评论(0)