深度学习函数
月韵清风
这个作者很懒,什么都没留下…
展开
-
线性函数预测
线性函数预测import tensorflow as tfimport numpy as npimport matplotlib.pyplot as pltx_data = np.random.rand(100).astype(np.float32)y_data = x_data*0.1 + 0.3Weights = tf.Variable(tf.random_uniform([1]...原创 2018-10-18 16:17:29 · 278 阅读 · 0 评论 -
nms
NMS 非极大值抑制import tensorflow as tfimport numpy as nprects=np.asarray([[1,2,3,4],[1,3,3,4], [1,3,4,4],[1,1,4,4],[1,1,3,4]],dtype=np.float32)scores=np.asarray([0.4,0.5,0.72,0.9,0.45], dtype=np.float...原创 2018-10-18 16:29:36 · 512 阅读 · 0 评论 -
huber loss
huber losshuber loss 是一种优化平方loss的一种方式,使得loss变化没有那么大。import numpy as npimport matplotlib.pyplot as pltimport tensorflow as tfdef huber_loss(x, d): x = np.abs(x) return ( x<=d ) * x**2...原创 2018-10-13 21:09:44 · 1314 阅读 · 0 评论 -
tf.where
tf.whereimport tensorflow as tftemp = tf.reshape(tf.range(0, 16) + tf.constant(1, shape=[16]), [4, 1, 2, 2])category_index = tf.where(tf.greater(temp, 6))with tf.Session() as sess: a = sess....原创 2018-10-18 22:23:21 · 182 阅读 · 0 评论 -
pytorch简单测试
pytorch demoimport torch.nn as nnimport torch.nn.functional as Ffrom torch.autograd import Variableimport torchimport torch.optim as optimclass Net(nn.Module):#需要继承这个类 def __init__(self): ...原创 2018-10-31 10:53:14 · 3010 阅读 · 0 评论 -
pytorch 分布式训练
pytorch 分布式训练参考文献https://pytorch.org/tutorials/intermediate/dist_tuto.htmldemoimport osimport torchimport torch.distributed as distfrom torch.multiprocessing import Processdef run(rank, size)...原创 2018-11-06 10:08:52 · 1917 阅读 · 0 评论 -
Pytorch Distributed 初始化
Pytorch Distributed 初始化方法参考文献https://pytorch.org/docs/master/distributed.html初始化torch.distributed.init_process_group(backend, init_method='env://', **kwargs)参数说明backend(str): 后端选择,包括 tcp mpi...原创 2018-11-06 11:15:16 · 5478 阅读 · 0 评论 -
pytorch 多gpu训练
pytorch 多gpu训练用nn.DataParallel重新包装一下device_ids=[0, 1, 2]model = model.cuda(device_ids[0])model = nn.DataParallel(model, device_ids=device_ids)optimizer = optim.SGD(midel.parameters(), lr=learning...原创 2018-11-05 10:59:24 · 271 阅读 · 0 评论 -
SEnet --se module
这是SEnet 的特征融合部分,import tensorflow as tffrom tflearn.layers.conv import global_avg_poolclass SE_layer(): def __init__(self, x, training=True): self.training = training def Global_A...原创 2018-11-16 16:52:36 · 1443 阅读 · 0 评论