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.

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

关于小数据集下设置aug=true后,会导致测试时outputs值整体小于0.5 #37

Closed
Fuzhiyuan opened this issue Jan 28, 2022 · 3 comments

Comments

@Fuzhiyuan
Copy link

我自己3d脊椎训练的数据集是7张nii.gz文件,epoch为20,batch为4.
设置aug=true后,进行推理时,手动输出outputs最大值小于0.5,导致result图像为空,但是设置false后就会正常(指result最大值会大于0.5)。
目前还没试验过其他数据集,这一点可以注意一下。

@Fuzhiyuan
Copy link
Author

Fuzhiyuan commented Jan 28, 2022

还有我在数据集1000张图片下训练100次,其余参数为默认值,没有开启aug,损失函数稳定到0.0007附近,但是推理的结果仍不理想,缩小patch_size是一个好的选择吗?准备改成256 time256 time 16

@MontaEllis
Copy link
Owner

这是一个小bug,现在已经修复,谢谢您的建议。

@Fuzhiyuan
Copy link
Author

Fuzhiyuan commented Oct 11, 2022 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants