本次将一个使用Pytorch的一个实战项目，记录流程：自定义数据集->数据加载->搭建神经网络->迁移学习->保存模型->加载模型->测试模型 自定义数据集 参考我的上一篇博客：自定义数据集处理 数据加载 默认小伙伴有对深度学习 ...

数据转换 1.numpy->tensor data2 = torch.tensor(data1) 2.tensor->numpy 非训练数据（训练结束后的tensor）的转换： data2 = np.array(data1) 如果是训练过程中需要转换，则： data2 ...

num_sequence.py """ 数字序列化方法 """ class NumSequence: """ input : intintint output :[i ...

今天训练keras时，发现在某些参数下，训练过程中的loss和acc在开始就很差（loss很大，acc很小，大概0.1左右）然后就稳定不变了，经过思考发现可能是步长设置的太大了，于是改变步长，小于默认值0.001，定为0.0005。效果显著提升。代码： train ...

目录 代码下载 更改配置及编译 测试初始权重 准备自己的数据集 划分测试集和训练集 将voc文件格式转为yolo文件格式（xml->txt） 训练 测试 1 代码下载 首先把代码下载下来，可以用下面命令下载 ...

在上一篇博客中，我们快速搭建，训练了一个小型的网络，但是存在一下问题。 仅仅是使用了 CPU，并没有使用 GPU 进行训练； 学习率太高，导致最后数值提不上去； 针对这2个问题，我们进行统一的解决。 并最后写一个 detect 模块，将我们写出的网络进行应用。 pytorch ...

一、背景 kaggle上有这样一个题目，关于盐份预测的语义分割题目。TGS Salt Identification Challenge | Kaggle https://www.kaggle.com ...

NaN 计算softmax loss时要用numeric robust 的计算方式. softmax与 loss可能要分开计算. 得到前者的计算方式可以是常规方法. 但计算后者时要注意无穷大和 ...