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

batchsize问题 #70

Open
MsRabbits opened this issue Jun 10, 2020 · 5 comments
Open

batchsize问题 #70

MsRabbits opened this issue Jun 10, 2020 · 5 comments

Comments

@MsRabbits
Copy link

作者您好 我按照您的代码自己复现了一遍 想请问如果我自己改写的话 batchsize想增加的话 是要从什么角度更改您的代码呢 谢谢您

@yangxue0827
Copy link
Owner

这个repo的效果不好,建议用这个FPN:https://github.com/DetectionTeamUCAS/FPN_Tensorflow

要改成多bs的话,数据的读取,roi的assign什么的,正负样本的采样等,很多地方都要修改。

@MsRabbits
Copy link
Author

这个repo的效果不好,建议用这个FPN:https://github.com/DetectionTeamUCAS/FPN_Tensorflow

要改成多bs的话,数据的读取,roi的assign什么的,正负样本的采样等,很多地方都要修改。

好的谢谢您的回复!我看您新发的这个版本的好像也是batchsize设置为1的,因为我的数据集有三万张图,我昨天试着跑了一天发现loss一直震荡,这个和batchsize太小是不是有关系呢,还有请请教您三万张训练集的话您这个代码我应该跑多少代才会有效果呢?

@yangxue0827
Copy link
Owner

20个epoch左右吧,一般第一次学习率下降之后的那个epoch效果趋于稳定。

@MsRabbits
Copy link
Author

20个epoch左右吧,一般第一次学习率下降之后的那个epoch效果趋于稳定。
好的 谢谢您回复。再想请教一下测试模型的话 是直接运行eval.py就可以了吗

@MsRabbits
Copy link
Author

我在运行eval.py的时候报了下面这个错误,请问这个应该是什么原因呢?
Traceback (most recent call last):
File "E:/Tree/sar_ship_detection_batch/FPN_TensorFlow/tools/eval.py", line 323, in
eval_dict_convert(args)
File "E:/Tree/sar_ship_detection_batch/FPN_TensorFlow/tools/eval.py", line 45, in eval_dict_convert
spatial_squeeze=False)
File "E:/Tree/sar_ship_detection_batch/FPN_TensorFlow/tools..\data\io../..\libs\networks\network_factory.py", line 49, in get_network_byname
with slim.arg_scope(resnet_v1.resnet_arg_scope(weight_decay=FLAGS.weight_decay)):
File "E:\anaconda\anaconda\envs\tensorflow\lib\site-packages\tensorflow\python\platform\flags.py", line 84, in getattr
wrapped(_sys.argv)
File "E:\anaconda\anaconda\envs\tensorflow\lib\site-packages\absl\flags_flagvalues.py", line 633, in call
name, value, suggestions=suggestions)
absl.flags._exceptions.UnrecognizedFlagError: Unknown command line flag 'weights'. Did you mean: weight_decay ?

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