设为首页
收藏本站
切换到宽版
登录
注册
找回密码
搜索
搜索
本版
文章
帖子
用户
快捷导航
易学开发平台
易学在线课堂
易学智能论坛
BBS
易学智能
»
易学智能论坛
›
管理运维
›
租赁平台
›
用tensorflow训练目标检测模型(数据集用的minist) ...
返回列表
查看:
2315
|
回复:
1
[远程开发]
用tensorflow训练目标检测模型(数据集用的minist)
[复制链接]
2919005896
2919005896
当前离线
积分
382
44
主题
139
帖子
382
积分
论坛管理
积分
382
发消息
发表于 2018-11-16 17:53:29
|
显示全部楼层
|
阅读模式
本帖最后由 2919005896 于 2018-11-19 08:04 编辑
1首先在官网下载minist数据集,数据集下载链接参考
http://yann.lecun.com/exdb/mnist/
2将下载好的数据集上传到易学智能开发平台上,下载链接参照:
https://bbs.easyaiforum.cn/thread-915-1-1.html
3我们参照的代码是github网站中的tensorflow下的official下的mnist的代码,代码网址是
https://github.com/tensorflow/models/tree/r1.5/official/mnist
,我们选择分支是r1.5,下载整个工程,然后解压,将解压的文件放在pycharm的工程目录下。
4,打开pycharm进行远程调试,pycharm远程调试的参考:
https://bbs.easyaiforum.cn/thread-625-1-1.html
5,数据处理:因为我们将数据集提前下载好了(程序中是需要下载的),所以我们需要将dataset.py的download函数进行修改,可以直接对路径进行判断,如图所示download函数的最终结果如下:
6,在minist.py中的class MNISTArgParser()中的self.add_argument()中的‘--date_dir’属性值改为第二步中存储在开发平台中的数据集的路径(即为刚刚上传到开发平台上的mnist数据集的路径),然后class MNISTArgParser()中的self.add_argument()中的‘--model_dir’属性值修模型创建的目录(自己决定)可以参考如下图:
7,然后点击运行mnist.py文件训练模型即可。训练结果如下图
:
8训练完模型之后运行mnist_test.py就可以得到测试结果了(可能会报错:module 'tensorflow.contrib.estimator' has no attribute 'TowerOptimizer',我们只要找到报错的这行代码删除就行(这个是关于多个gpu的我们这里没有用到)))。
回复
使用道具
举报
shaoheshaohe
shaoheshaohe
当前离线
积分
6670
665
主题
1234
帖子
6670
积分
xdtech
积分
6670
发消息
发表于 2018-11-16 21:56:12
|
显示全部楼层
写的还不错
就是
好像有些文字错误
回复
使用道具
举报
返回列表
高级模式
B
Color
Image
Link
Quote
Code
Smilies
您需要登录后才可以回帖
登录
|
立即注册
本版积分规则
发表回复
回帖后跳转到最后一页
快速回复
返回顶部
返回列表