Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

数据集包含不同尺寸图片训练报错 #105

Open
1404561326521 opened this issue Sep 19, 2024 · 2 comments
Open

数据集包含不同尺寸图片训练报错 #105

1404561326521 opened this issue Sep 19, 2024 · 2 comments

Comments

@1404561326521
Copy link

自定义数据集中包含不同尺寸的图片，并且也不是8的倍数，训练时会报错：
image
这个有什么解决办法吗?难道需要先对所有数据手动裁剪在进行训练吗？实际情况中输入的每张图片尺寸也不会都是一样的啊
@GUIMINLONG
Copy link

自定义数据集中包含不同尺寸的图片，并且也不是8的倍数，训练时会报错： image 这个有什么解决办法吗?难道需要先对所有数据手动裁剪在进行训练吗？实际情况中输入的每张图片尺寸也不会都是一样的啊

就是要裁剪的吧，模型的框架确定，允许的输入的维度就定死了的。实际情况应该是大了脚本裁剪，小了脚本扩充的吧。

@1404561326521
Copy link
Author

自定义数据集中包含不同尺寸的图片，并且也不是8的倍数，训练时会报错： image 这个有什么解决办法吗?难道需要先对所有数据手动裁剪在进行训练吗？实际情况中输入的每张图片尺寸也不会都是一样的啊

就是要裁剪的吧，模型的框架确定，允许的输入的维度就定死了的。实际情况应该是大了脚本裁剪，小了脚本扩充的吧。

因为cityscapes数据集尺寸都是统一为2048×1024的，所以源代码可以直接运行，但是在自定义数据上需要自己写一个数据预处理的过程把不同尺寸的的图片缩放到统一尺寸再送入网络就可以了，这样就解决了上述问题

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@GUIMINLONG @1404561326521