search

raytroop / YOLOv3_tf

Try to implement YOLOv3 with TensorFlow
67 stars 20 forks source link

你是重新训练生成的ckpt模型，还是将yolov3.weights转换为ckpt模型？ #3

Closed GitEasonXu closed 5 years ago

GitEasonXu commented 6 years ago

Github上有yolov3.weights to Keras模型的代码，但是一直找不到Tensorflow模型的实现，实在感谢。 如果我想要使用预训练模型进行再训练应该如何进行？还望指点。

raytroop commented 6 years ago

I DON'T really convert darknet weights to tensorflow ckt, instead extract feature extractor weights darknet53.conv.74 format of darknet and save it as npz using save_npz.py (modified from yad2k) This way, tensorflow weights can be initialized with them. I use train_net.py to train the following neural network(head and detector), feature extractor is frozen at the time.

GitEasonXu commented 6 years ago

谢谢你的回复,