![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow
Serrie.
心存敬畏
展开
-
faster中的NMS计算以及理解(tensorflow)
# -*- coding:utf-8 -*-import numpy as npdef py_cpu_nms(dets, thresh): """ :param dets:N*M 二维数组, N是BBOX的个数, M的前四位对应的是(x1, y1, x2, y2) 第5位是对应的分数 x y为坐标 :param thresh:0.3 0.5.... :r...原创 2019-04-12 09:04:51 · 997 阅读 · 1 评论 -
超分辨率网络ESPCN中的pixel shuffle--几种代码思路(基于TF, Pytorch)
方法一:源码来自github:https://github.com/JuheonYi/VESPCN-tensorflow中 ESPCN部分首先简单的来看ESPCN的网络结构搭建 conv--conv--conv--ps def network(self, LR): feature_tmp = tf.layers.conv2d(LR, 64, 5, stri...原创 2019-07-10 11:46:32 · 4547 阅读 · 1 评论 -
tensorflow中的thread(queue以及Coordinator)的使用与控制
import tensorflow as tfimport numpy as npimport threadingimport time# def MyLoop(coord, worker_id):# while not coord.should_stop():# if np.random.rand() * 1000 % 5 > 3:# ...原创 2019-04-13 13:49:54 · 665 阅读 · 0 评论 -
使用torch以及tensorflow训练一个最简单网络的基本步骤
torch:import torchimport torch.nn.functional as Fimport matplotlib.pyplot as pltx = torch.Tensor.unsqueeze(torch.Tensor.linspace(-1, 1, 100), dim=1) # x data (tensor), shape=(100, 1)y = x.po...原创 2019-04-12 09:19:38 · 696 阅读 · 0 评论 -
tensorflow中参数解析以及使用
#method1 sysimport sysgpus = sys.argv[1]batch_size = sys.argv[2]print(gpus)print(batch_size)#method2 argparse 输入参数时要输入 -gpu=1import argparseparser = argparse.ArgumentParser(description='ma...原创 2019-04-12 09:18:14 · 504 阅读 · 0 评论 -
基础Estimator以及自定义Estimator
基础Estimator#--coding:utf-8--import numpy as npimport tensorflow as tffrom tensorflow.examples.tutorials.mnist import input_datatf.logging.set_verbosity(tf.logging.INFO)mnist = input_data.read...原创 2019-04-12 09:17:15 · 1027 阅读 · 0 评论 -
Inception-v3网络结构搭建(基于tensorflow-slim)
#-*- coding:utf-8 -*-import tensorflow as tf#加载slim库slim = tf.contrib.slim#slim.arg_scope()函数可以用于设置默认的参数取值。此函数的第一个参数是一个函数列表,在这个列表中的函数将使用默认的参数设置。with slim.arg_scope([slim.conv2d, slim.max_pool2...原创 2019-04-12 09:01:56 · 679 阅读 · 0 评论 -
ResNet 以及DenseNet 网络结构搭建(tensorflow)
ResNet:ResNet:#-*- coding:utf-8 -*-import tensorflow as tffrom __future__ import absolute_import #将新版本的特性引进当前版本,也就是说我们可以在当前版本使用新版本的一些特性__BATCH_NORM_DECAY = 0.997__BATCH_NORM_EPSILON = 1e-...原创 2019-04-12 09:03:01 · 995 阅读 · 1 评论 -
tensorflow下TFRecord的读与写
读:TFRecordRead:import tensorflow as tf#读取train.tfrecord中的数据def read_and_decode(filename): #创建一个reader来读取TFRecord文件中的样例 reader = tf.TFRecordReader() #创建一个队列来维护输入文件列表 file_queue ...原创 2019-04-12 09:09:21 · 198 阅读 · 0 评论 -
TensorFlow模型冻结(ckpt转为pb)笔记
# -*- coding:utf-8 -*-""" This file used to freeze tensorflow .ckpt to .pb"""import tensorflow as tf#两种方式 方法1:函数方法,传入sessiondef freeze_session(session, keep_var_name=None, output_names=None...原创 2019-07-06 16:13:20 · 2088 阅读 · 0 评论