源码

报告

心得

具体介绍可以看报告,这里就主要讲讲这次实操的一些心得。

文件结构

个人推荐如下的结构(也可以继续细分)。

mainFolder/
    datasetPath/
        dataset/
            train/
                info.txt
            val/
                info.txt
            info.txt
    modelPath/
        *1.pth
        *2.pth
        ...
        *10.pth
    pyPath/
        model.py
        dataset.py
        imagePreProcess.py
    config.py
    train.py
    val.py

config.py

将全局设置独立开可以极为方便的进行调参工作。

pyPath/

其中可以储存 train.py 中需要的所有东西,包括但不限于 Model, Dataset, Optim, Loss, Transforms …

datasetPath/

储存各种数据集,方便切换训练不同数据集。

模型搭建

原本是想自己搭模型的,这样可扩展性强一些,不过确实由于自己的水平有限,还没有真正理解 Pytorch 的底层实现,日后学会了肯定会自己搭建模型。