查看: 1688|回复: 1

[远程开发] 用tensorflow训练目标检测模型(数据集用的minist)

[复制链接]

44

主题

139

帖子

382

积分

论坛管理

Rank: 4

积分
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的工程目录下。
图1.png

4,打开pycharm进行远程调试,pycharm远程调试的参考:https://bbs.easyaiforum.cn/thread-625-1-1.html

5,数据处理:因为我们将数据集提前下载好了(程序中是需要下载的),所以我们需要将dataset.py的download函数进行修改,可以直接对路径进行判断,如图所示download函数的最终结果如下:
图2.png

6,在minist.py中的class MNISTArgParser()中的self.add_argument()中的‘--date_dir’属性值改为第二步中存储在开发平台中的数据集的路径(即为刚刚上传到开发平台上的mnist数据集的路径),然后class MNISTArgParser()中的self.add_argument()中的‘--model_dir’属性值修模型创建的目录(自己决定)可以参考如下图:
图3.png

7,然后点击运行mnist.py文件训练模型即可。训练结果如下图
图4.png
8训练完模型之后运行mnist_test.py就可以得到测试结果了(可能会报错:module 'tensorflow.contrib.estimator' has no attribute 'TowerOptimizer',我们只要找到报错的这行代码删除就行(这个是关于多个gpu的我们这里没有用到)))。
图5.png







回复

使用道具 举报

665

主题

1234

帖子

6575

积分

xdtech

Rank: 5Rank: 5

积分
6575
发表于 2018-11-16 21:56:12 | 显示全部楼层
写的还不错
就是
好像有些文字错误
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表