发布于2020-02-26 11:18 阅读(2005) 评论(0) 点赞(7) 收藏(1)
注意:efficientnet这个库在7月24的时候更新了,keras和tensorflow.keras框架也可以用,
想要学习EfficientNet,如果你要训练的模型是7月24日之前的,请安装0.0.4版本。安装代码: pip install -U efficientnet==0.0.4 -i https://pypi.tuna.tsinghua.edu.cn/simple
环境:tensorflow > 1.12.0、Keras > 2.2.0 、keras_applications > 1.0.7
安装方法:参考链接
1、从github上安装(一般不推荐,网太慢)
pip install -U git+https://github.com/qubvel/efficientnet
2、安装稳定版(-i 后边是清华提供的镜像文件,速度飞快)
pip install -U efficientnet -i https://pypi.tuna.tsinghua.edu.cn/simple
3、安装最新版
pip install -U --pre efficientnet -i https://pypi.tuna.tsinghua.edu.cn/simple
为什么EfficientNet是最好的分类网络、什么是EfficientNet?
自己训练这个网络有点不切实际,肯定是用迁移学习进行分类任务的。下面先罗列我参考过的一些文章,如果是在tensorflow、keras环境下,则简称tf-k环境下。
tf-k环境下:
pytorch环境下:
本代码用来参加了AI研习社举办的一个分类竞赛,竞赛地址https://god.yanxishe.com/26。第一次用EfficientNet效果还不错,正确率达到95%。
我的代码运行在kaggle提供的GPU环境下。kaggle每周都会免费提供30个小时的GPU训练时常,对穷学生来说很实用。
闲话不多说,正式介绍代码:
因为efficientnet这个库在7月24号更新过,更新后tf-k环境下也可以用。如果你要用更新前的版本则需安装efficientnet==0.0.4版本。并且使用方法也有所不同:
老版本:
- #导入
- from efficientnet import EfficientNetB0 as Net
-
- #使用
- conv_base = Net(weights='imagenet', include_top=False, input_shape=input_shape)
新版本:
- #导入
- import efficientnet.tfkeras as efn
- #使用
- base_model = efn.EfficientNetB0(
- input_shape=(224,224,3),
- wights='imagenet',
- include_top=False)
竞赛的全部代码(不需要积分就能下载):https://download.csdn.net/download/jiachang98/12167199
作者:骏马
链接:https://www.pythonheidong.com/blog/article/233941/dd7ddf6bdfec3f4866dd/
来源:python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容:(最多支持255个字符)
---无人问津也好,技不如人也罢,你都要试着安静下来,去做自己该做的事,而不是让内心的烦躁、焦虑,坏掉你本来就不多的热情和定力
Copyright © 2018-2021 python黑洞网 All Rights Reserved 版权所有,并保留所有权利。 京ICP备18063182号-1
投诉与举报,广告合作请联系vgs_info@163.com或QQ3083709327
免责声明:网站文章均由用户上传,仅供读者学习交流使用,禁止用做商业用途。若文章涉及色情,反动,侵权等违法信息,请向我们举报,一经核实我们会立即删除!