tensorflow
WTHunt
厚积薄发
展开
-
resnet50训练imagenet记录
参考资料:resnet50结构:https://www.jianshu.com/p/993c03c22d52resnet50训练条件:百度paddleimagenet labels:https://raw.githubusercontent.com/soumith/imagenetloader.torch/master/valprep.shResNet50模型的复现详解原创 2021-04-01 21:04:02 · 2336 阅读 · 0 评论 -
resnet50训练
from __future__ import absolute_importfrom __future__ import divisionfrom __future__ import print_functionimport argparseimport sysimport tempfilefrom resnet50 import create_modelfrom solver import create_admm_solverfrom TFRecord_data_gen import .原创 2021-03-29 10:17:55 · 741 阅读 · 2 评论 -
Tensorflow—tfrecord数据集生成与使用
参考内容:数据读取的官方教程:Tensorflow导入数据以及使用数据tfrecord数据集生成:数据准备:图片数据+图片目录与label一一对应的的txt先读取图片信息的txt文件,得到每个图片的路径以及它们的标签,然后对这个图片作一些预处理,最后将图片以及它对应的标签序列化,并建立图片和标签的索引(即以下代码的”img_raw”, “label”)。详见代码。import randomimport tensorflow as tffrom PIL import Imag..原创 2021-03-03 17:37:47 · 751 阅读 · 1 评论 -
TensorFlow-Slim API 官方教程
转载链接:https://blog.csdn.net/wanttifa/article/details/90208398转载 2021-02-19 11:43:27 · 99 阅读 · 1 评论 -
TF设置线程数
如果用服务器,tensorflow一旦train起来,别人就别想用了,解决方法也很简单,设置一下线程数就可以了cpu_num = 16config = tf.ConfigProto(device_count={"CPU": cpu_num}, inter_op_parallelism_threads = cpu_num, intra_op_parallelism_threads = cpu_num, log_device_placement=Tr原创 2020-11-24 13:00:36 · 1162 阅读 · 0 评论 -
TF实现多minibatch梯度累加及反向更新
参考链接:TF中optimizor源码:https://blog.csdn.net/Huang_Fj/article/details/102688509如何累加梯度进行反向:https://stackoverflow.com/questions/46772685/how-to-accumulate-gradients-in-tensorflowhttps://blog.csdn.net/weixin_41560402/article/details/106930463问题背景及解决:原创 2020-11-02 12:21:33 · 1254 阅读 · 1 评论 -
如何插入8bit量化节点(tensorflow)
目录tf流图graph基础知识默认图创建显式图创建多个图调用tf伪量化接口插入kernel、层间量化节点tf流图graph基础知识默认图import tensorflow as tfimport numpy as npa = tf.constant(123)print(a.graph)print(tf.get_default_graph())输出...原创 2019-12-08 21:48:48 · 1177 阅读 · 2 评论 -
smooth l1(huber)+binary cross entropy详解(tensorflow+pytorch)
前言上篇讲到yolov2 loss中使用了两个loss分别时wh使用的smoothl1loss以及其他的BCEloss,这里做个扩展介绍,并对比pytorch以及tensorflow的api接口的使用smoothl1(huber loss)转载自知乎高赞回答:https://www.jianshu.com/p/19483787fa24smooth L1 loss能从两个方面限...原创 2019-12-01 10:26:55 · 1046 阅读 · 0 评论 -
优图yolo-v2 loss解析(tensorflow)
前言最近在做yolo检测模型的8bit落地工作,因需求手撸了优图的yolov2的loss,做个记录方便日后查看。本篇只介绍核心的loss部分,其余详细的代码放到github上面去了,详见https://github.com/XhtZz/yolo-v2-lossloss详解def yolo_loss( target, output, ...原创 2019-11-29 17:17:28 · 207 阅读 · 0 评论 -
softmax简单模型处理Mnist分类问题
import tensorflow as tfimport input_data#导入数据mnist = input_data.read_data_sets('MNIST_data',one_hot=True)#权重及偏置设置weights = tf.Variable(tf.zeros([784,10]))biases = tf.Variable(tf.zeros([10]))...原创 2018-12-16 00:39:49 · 307 阅读 · 0 评论 -
Tensorflow入门
Tensorflow初探:需要理解的基本问题: 如何将计算流程表示成图? 怎样通过session来执行图计算? 如何将数据表示成tensors? 如何用Variables来保持状态信息? 如何分别使用feeds和fetches来填充数据和抓取任意的结果? 1. TensorFlow 是什么是一个深度学习库,由 Google 开源,可以对定义...原创 2018-09-09 18:05:58 · 211 阅读 · 0 评论