TensorFlow2.0-Examples icon indicating copy to clipboard operation
TensorFlow2.0-Examples copied to clipboard

YOLOv3中训练自己的数据集时可以调用官方的yolov3.weights,在它的基础上进行训练吗?

Open monoloxo opened this issue 4 years ago • 8 comments

非常感谢有这样一份2.0版本的示例,有一个问题是我在训练自己的数据集时初始loss很大,最终也没得到一个理想的效果,所以想请教如何在yolov3.weights的基础上进行训练?

monoloxo avatar Mar 31 '20 10:03 monoloxo

同问,训练自己的模型的时候,backbone 部分的权重为啥不直接加载官方已经训练好的权重呢

forever208 avatar May 30 '20 14:05 forever208

可以的,但需要在加载模型的时候注释掉原有的模型加载方式,使用如45行的方式加载

------------------ 原始邮件 ------------------ 发件人: "forever208"<[email protected]>; 发送时间: 2020年5月30日(星期六) 晚上10:29 收件人: "YunYang1994/TensorFlow2.0-Examples"<[email protected]>; 抄送: "Subscribed"<[email protected]>; 主题: Re: [YunYang1994/TensorFlow2.0-Examples] YOLOv3中训练自己的数据集时可以调用官方的yolov3.weights,在它的基础上进行训练吗? (#87)

同问,训练自己的模型的时候,backbone 部分的权重为啥不直接加载官方已经训练好的权重呢

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or unsubscribe.

coyote000 avatar May 30 '20 14:05 coyote000

可以的,但需要在加载模型的时候注释掉原有的模型加载方式,使用如45行的方式加载 ------------------ 原始邮件 ------------------ 发件人: "forever208"<[email protected]>; 发送时间: 2020年5月30日(星期六) 晚上10:29 收件人: "YunYang1994/TensorFlow2.0-Examples"<[email protected]>; 抄送: "Subscribed"<[email protected]>; 主题: Re: [YunYang1994/TensorFlow2.0-Examples] YOLOv3中训练自己的数据集时可以调用官方的yolov3.weights,在它的基础上进行训练吗? (#87) 同问,训练自己的模型的时候,backbone 部分的权重为啥不直接加载官方已经训练好的权重呢 — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or unsubscribe.

Hello,45行的方式具体指的是什么? 方便帮忙解答一下嘛?感谢!

forever208 avatar May 30 '20 14:05 forever208

可以在train.py 'model = tf.keras.Model(input_tensor, output_tensors)'这句话后面(在40行左右)加上 model.load_weights("./yolov3_myself")

llllala avatar Jun 07 '20 05:06 llllala

感谢 llllala,另外想再问一下,加载权重后准备训练,我想在训练阶段 freeze bn 层,请问需要咋操作呢。bn 的子类我看得不太明白,还请赐教。感恩~!

forever208 avatar Jun 07 '20 08:06 forever208

您好,您添加model.load_weights("./yolov3_myself")是否正常运行,我这边添加完后会提示以下错误 ./yolov3.weights: Data loss: not an sstable (bad magic number): perhaps your file is in a different file format and you need to use a different restore operator?

Mr-yun avatar Jun 30 '20 14:06 Mr-yun

可以在train.py 'model = tf.keras.Model(input_tensor, output_tensors)'这句话后面(在40行左右)加上 model.load_weights("./yolov3_myself")

您好,model.load_weights("./yolov3.weights"),显示Unable to open file (file signature not found)改怎么解决

wangyaxin1998 avatar Apr 08 '21 06:04 wangyaxin1998

可以在train.py 'model = tf.keras.Model(input_tensor, output_tensors)'这句话后面(在40行左右)加上 model.load_weights("./yolov3_myself")

这样就可以了吗?还需要修改别的地方吗?

mastergao57 avatar Mar 22 '22 10:03 mastergao57