机器学习框架实操(TF/PT/Keras/Caffe)
Loy_Fan
Life can not be planned
展开
-
torchvision.transform图片转变/数据增强方法总结
来自官方文档-torchvision/transformstorchvision.transformCrop裁剪Flip and Rotation翻转和旋转图片变换对transform的后续操作Crop裁剪transforms.CenterCrop中心裁剪torchvision.transforms.CenterCrop(size)根据给定的size从中心裁剪,size可以是seque...原创 2020-04-17 14:16:28 · 3918 阅读 · 0 评论 -
PyTorch学习率衰减函数
PyTorch学习率衰减函数使用原创 2020-02-05 12:07:25 · 752 阅读 · 0 评论 -
PyTorch预训练模型保存与加载
PyTorch使用自定义层预训练模型后,加载模型需要包含当时定义的网络结构原创 2020-02-03 16:20:37 · 1969 阅读 · 1 评论 -
PyTorch的Tensor相关函数/操作
view改变张量的形状,相当于numpy的reshape。>>> a.shapetorch.Size([3, 2, 2])>>> a.view(3,4).shapetorch.Size([3, 4])permute给张量的维度换位置,期中的参数就是原来的维度的编号。>>> a.shapetorch.Size([3, 2, 2...原创 2020-01-01 22:05:28 · 473 阅读 · 0 评论 -
Keras使用TensorBoard可视化训练过程
添加关键代码在模型训练语句中添加回传训练信息的代码# 每隔一个训练循环就用柱状图显示信息callbacks = [keras.callbacks.TensorBoard(log_dir='自定义日志保存目录'.format(模型名称))]history = fashion_model.fit(训练集, 标签, epochs, batch_size, validation_split = 0...原创 2019-07-06 12:54:25 · 1981 阅读 · 0 评论 -
Keras自动下载的数据集/模型存放位置
Mac# 数据集~/.keras/datasets/# 模型~/.keras/models/Linux# 数据集~/.keras/datasets/Windows# win10C:\Users\user_name\.keras\datasets原创 2019-05-08 10:03:23 · 3959 阅读 · 0 评论 -
Torch的损失函数总结
nn.L1Loss取预测值和真实值的绝对误差的平均数。loss(x,y)=1N∑i=1N∣x−y∣\operatorname{loss}(\mathbf{x}, \mathbf{y})=\frac{1}{N} \sum_{i=1}^{N}|\mathbf{x}-\mathbf{y}|loss(x,y)=N1i=1∑N∣x−y∣nn.SmoothL1Loss也叫作 Huber Lo...原创 2019-08-17 10:55:29 · 1524 阅读 · 1 评论 -
Convolutional Neural Networks with TensorFlow
article linkAditya SharmaMarch 10th, 2018In this tutorial, you’ll learn how to construct and implement Convolutional Neural Networks (CNNs) in Python with the TensorFlow framework.TensorFlow is a...转载 2019-04-26 22:25:55 · 641 阅读 · 0 评论