- 博客(5)
- 收藏
- 关注
RSS订阅原创 若文件夹存在就创建文件夹否则删除重建
import osimport shutil#若模型保存文件夹不存在，创建模型保存文件夹，若存在，删除重建def mkr(path): if os.path.exists(path): shutil.rmtree(path) os.mkdir(path) else: os.mkdir(path)if __name__ == ...
2019-10-31 11:34:28
820
原创 torch.cat()张量拼接
torch.cat((a,b),dim),两个向量在dim维度上拼接,要求被拼接的矩阵在另一个维度相等import torchA=torch.ones(2,3) #2x3的张量（矩阵）B=2*torch.ones(4,3)#4x3的张量（矩阵）C=torch.cat((A,B),0)#按维数0（行）拼接，要求两个矩阵另一个维度（列）相等print(C)## tensor([[1.,...
2019-10-30 15:08:59
8369
原创 retinanet 中的filter_detections，为何要stack上labels
看代码，若网络预测出了10个box=[b1,b2,b3,,b10]对应10个scores=·[s1,s2,s3,...s10]=[0.1,0.2,0.3,0.4,..., 1],10个labels=[0,1,2,3,4,0,1,2,0,2] #就是网络检测种类有5种目标step1 ,对于scores,过滤出得分大于阈值=0.05的那些box,记录其索引indices=[5,6,7,8,9]...
2019-10-16 15:08:01
251
原创 pytorch 安装
1。anaconda 创建一个虚拟环境，虚拟环境下（pytorch 1.2 cuda 10 cudnn 7.6)2。 conda install pytorch=1.2.0 cuda100 -c pytorch 若安装过程中，遇到某一个包比较大，重新运行上面命令，会把上次下载失败的包重新下载一下，下载好的就不下载了3。备注：我用了pip install pytorch 但在用gp...
2019-10-15 16:44:26
189
原创 tf.truncated_normal(),tf.pad()及其在卷积层的应用
tf.truncated_normal(shape,mean=0.0,stddev=1.0,dtype=tf.float32,seed=None,name=None)函数功能:生成均值=mean=0,方差=stddev=1的正态分布矩阵,矩阵维度=shape,矩阵元素值范围[mean-2*stdeev, mean+2*stddev]例子:import tensorflow as t...
2019-10-08 15:20:48
377
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人