qq249356520的博客

# -*- 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...

方法一：源码来自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...

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:# ...

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...

#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...

基础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...

#-*- coding:utf-8 -*-import tensorflow as tf#加载slim库slim = tf.contrib.slim#slim.arg_scope()函数可以用于设置默认的参数取值。此函数的第一个参数是一个函数列表，在这个列表中的函数将使用默认的参数设置。with slim.arg_scope([slim.conv2d, slim.max_pool2...

ResNet：ResNet：#-*- coding:utf-8 -*-import tensorflow as tffrom __future__ import absolute_import #将新版本的特性引进当前版本，也就是说我们可以在当前版本使用新版本的一些特性__BATCH_NORM_DECAY = 0.997__BATCH_NORM_EPSILON = 1e-...

读：TFRecordRead：import tensorflow as tf#读取train.tfrecord中的数据def read_and_decode(filename): #创建一个reader来读取TFRecord文件中的样例 reader = tf.TFRecordReader() #创建一个队列来维护输入文件列表 file_queue ...

# -*- 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...