深度学习训练遇到的坑（一）

如果使用的训练集图像的分辨率很高（几千x几千的那种），不要在训练的过程中将resize图像方法封装到transforms.Compose中，这样会使得resize图像的过程变得特别慢，最好的方法是在训练之前就对所有图像做预处理，利用opencv的resize方法对训练集和验证集图像全部裁剪成特定尺寸，得到低分辨率的图像后再进行训练。...

Legolas~

440人浏览 · 2021-05-08 14:06:51

Legolas~ · 2021-05-08 14:06:51 发布

如果使用的训练集图像的分辨率很高（几千x几千的那种），不要在训练的过程中将resize图像方法封装到transforms.Compose中，这样会使得resize图像的过程变得特别慢，最好的方法是在训练之前就对所有图像做预处理，利用opencv的resize方法对训练集和验证集图像全部裁剪成特定尺寸，得到低分辨率的图像后再进行训练。