faster-rcnn-keras icon indicating copy to clipboard operation
faster-rcnn-keras copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

您好，请问这个模型的初始权重(voc_weights.h5)文件可以换成其他网络吗？比如换成vgg16或者resnet101什么的。

Open Eve66666 opened this issue 5 years ago • 3 comments

您好，我有几个问题想请教您！ 1.请问这个模型的初始权重(voc_weights.h5)文件可以换成其他网络吗？比如换成vgg16或者resnet101什么的。

2.您设定的epoch为100个，每个epoch2000(step?)次，我的电脑(win10.1050ti.4G)一个epoch训练时间大概要半个小时。之前我在琢磨git上的Faster-RCNN-TensorFlow-Python3，他迭代一次所需的时间只有1-2s(我上个厕所就迭代了一千多次。。。但后面他的demo.py文件我运行起来无法生成预测的图片，怎么改都不行，加上训练所需时间实在太短，感觉不真实，所以就放弃研究那个模型了。)请问您有研究过那个模型吗？为什么两个模型的训练时间相差这么大？是我电脑环境的问题吗

我用的 tf-gpu 1.12.0 cudn 9.0.176 cudnn 7.3.0.29

Eve66666 avatar Mar 09 '20 02:03 Eve66666

1、我不太了解怎么把权重换成网络……我怀疑你可能搞错了 2、他的一个世代就是我的一个STEP吧？我没用过

bubbliiiing avatar Mar 09 '20 02:03 bubbliiiing

emmm，抱歉，第一个问题我重新描述一下= = 能否把您一开始给出的voc_weights.h5文件换成经过其他网络(比如ResNet101，VGG16)训练后得到的权重进行替换？

Eve66666 avatar Mar 09 '20 03:03 Eve66666

你可以换，但是你要网络结构也对应

bubbliiiing avatar Mar 09 '20 03:03 bubbliiiing