CNN卷积神经网络模型-02.调参(过拟合）

1、环境

运行环境

I512400+GPU3080+32GB

python310+torch-gpu版本

2800张图片（4类），2250训练，550验证

2、可视化观察

仅仅感受图像卷积后变化，调参帮助不大

第0次，未卷积原始图像：

第29次卷积训练结果：

 

 自动保存每次卷积图片

def filter_show(filters,l_c,n_epoch,n_batch):   
   if n_batch == 0:
        nx=8
        margin=3
        scale=10
        dt = filters.cpu().detach().numpy()
        FN, C, FH, FW = dt.shape
        ny = int(np.ceil(FN / nx)) 
        spath = './pic/'+str(n_epoch)+'_'+l_c+'/'
        if os.path.exists(spath)=