tensorflow
三寸光阴___
这个作者很懒,什么都没留下…
展开
-
google colab解决不能下载tensorflow_datasets下载数据集
前提:可以访问google,本地利用tensorflow_datasets下载数据集太慢参考链接:https://www.zhihu.com/question/362309074、https://www.jarvis73.com/2020/11/25/Tensorflow-Datasets/以cifar10为例1.打开以下网址,登录google账号:https://colab.research.google.com/drive/1RwYB7rQHfDyJZXBGE_nT4eKprpp3rpcU?us原创 2021-01-28 17:51:23 · 1310 阅读 · 0 评论 -
DoreFa-Net低bit量化学习
参考文章:https://zhuanlan.zhihu.com/p/163707781tensorflow实现:https://github.com/tensorpack/tensorpack/tree/master/examples/DoReFa-Netpytorch实现:https://github.com/666DZY666/model-compression原创 2020-09-29 18:08:38 · 439 阅读 · 0 评论 -
tensorflow模型裁剪资料收集
model-optimization官方:https://github.com/tensorflow/model-optimizationhttps://github.com/tensorflow/model-optimization/blob/master/tensorflow_model_optimization/g3doc/guide/pruning/pruning_with_keras.ipynbtensorflow.contrib.model_pruning.python剪枝模块:from原创 2020-08-21 17:00:10 · 447 阅读 · 0 评论 -
tensorflow lite将ssd_mobilenet移植至移动端
准备工作下载tensorflow源码git clone https://github.com/tensorflow/tensorflow.git下载tensorflow modelgit clone https://github.com/tensorflow/models.git下载并安装bazel编译工具https://docs.bazel.build/versions/mast...原创 2019-08-09 10:27:27 · 1418 阅读 · 3 评论 -
天池2019 年县域农业大脑AI挑战赛
发现比赛的时候已经只剩一个多星期了,最后勉强排在120名,做一下记录。数据训练:4716150141,7747046050测试:3724119903,2593628832首先训练要对训练集进行切割,在预测的时候要先切割再预测,再合并。训练集切割和测试集切割不一样,训练集切割的图像可以有交集,如果测试集图像有交集切割不太方便合并,故分为两部分代码。代码都是由参加比赛的大神们写的,只是做了简...原创 2019-08-04 18:21:40 · 1010 阅读 · 10 评论 -
faster rcnn inception_resnet_v2物品辨识比赛demo记录
使用tensorflow model里面的object detection训练的,因为没有时间限制,使用faster rcnn inception_resnet_v2识别10个类别,在1080ti上训练了5个小时,在1050上测试的,2s一张图片。定义 pascal_label_map.pbtxtitem { id: 1 name: 'cola'}item { id: 2 ...原创 2019-06-03 11:14:35 · 1774 阅读 · 0 评论 -
Slim MoileNetv2分类模型
主要参考这个项目并对其修改,以flowers为例:完整代码发在了github上先准备数据集:pic/train、pic/validation。运行:python data_convert.py -t pic/ \ --train-shards 2 \ --validation-shards 2 \ --num-threads 2 \ --dataset-name flower...原创 2019-03-27 18:40:39 · 218 阅读 · 4 评论 -
tensorflow ckpt模型和pb模型获取节点名称,以及ckpt转pb模型
ckptfrom tensorflow.python import pywrap_tensorflow checkpoint_path = 'model.ckpt-8000' reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) var_to_shape_map = reader.get_variable_to_sh...原创 2019-03-27 12:01:13 · 4174 阅读 · 18 评论 -
tensorflow框架使用小结
直接上代码,mnist手写数字识别:训练代码:from tensorflow.examples.tutorials.mnist import input_dataimport tensorflow as tfmnist = input_data.read_data_sets('MNIST_data', one_hot=True)sess = tf.InteractiveSession(...原创 2019-03-29 21:06:02 · 298 阅读 · 0 评论 -
C++传输图片到python
开发环境:VS2015,python3.6release版本编译,debug版本 import_array()会有错误,;PyArray_SimpleNewFromData()必须引入import_array(),我的头文件#include <numpy/arrayobject.h>,在 F:\Anaconda\envs\python36\Lib\site-packages\nump...原创 2019-02-27 11:42:25 · 5910 阅读 · 7 评论