深度学习
文章平均质量分 63
爱吃肉c
关关难过关关过,前路漫漫亦灿灿
展开
-
ONNX
首先将自己训练出来的pth模型转化为ONNX模型。之后将在网站中打开该onnx文件 即可看到网络模型。这是我自己的,改为自己的即可。原创 2024-03-21 16:42:46 · 190 阅读 · 0 评论 -
Dataset与DataLoader、transform
重写transform的目的:可以接受多个参数,可以保证对我们的图像和标注进行同步处理,比如图像分类任务,如果我们对图像及进行了预处理,比如进行了图像裁剪和缩放以及旋转等,其对应的标注框也应该做同步变换,否则就会出错,这时候就需要我们重写transform,对图像和标注做同步处理。数据增强可以增加训练集的样本数量,缓解过拟合,并提高模型的泛化能力,从而有效提升算法的性能。,因为读取的数据包括image和targets,不能直接使用默认的方法合成batch。1、将图像转换成tensor 的数据格式。原创 2024-03-19 15:04:37 · 658 阅读 · 0 评论