search

chuanenlin / drone-net

https://towardsdatascience.com/tutorial-build-an-object-detection-system-using-yolo-9a930513643a
152 stars 66 forks source link

Weight file and cfg file type #5

Open ngingihy opened 5 years ago

ngingihy commented 5 years ago

Hi, I'm wondering what pre-trained weight you used for training to generate the yolo-drone-weights, was it yolov3-tiny/yolov2/voc?

Also, did you do any pre-processing for the images before training?

chuanenlin commented 5 years ago

@ngingihy I used yolov3-tiny.cfg and weights from darknet53 (pre-trained on ImageNet). I didn't do any pre-processing before training.

ngingihy commented 5 years ago

So can i use your yolo-drone.weights as a pre-trained weight to train it more on far away drones images, with the yolo-drone.cfg file? or you think its better to start with yolo-tiny and darknet53 weights?

On Thu, Mar 28, 2019 at 12:36 AM Chuan-en Lin 林傳恩 notifications@github.com wrote:

@ngingihy https://github.com/ngingihy I used yolov3-tiny.cfg and weights from darknet53 https://pjreddie.com/darknet/imagenet/#darknet53 (pre-trained on ImageNet). I didn't do any pre-processing before training.

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub https://github.com/chuanenlin/drone-net/issues/5#issuecomment-477440294, or mute the thread https://github.com/notifications/unsubscribe-auth/AuvQRJQ6YBWFegRSV229s37CizxNtYGgks5vbECdgaJpZM4cPDWn .

ngingihy commented 5 years ago

So can i use your yolo-drone.weights as a pre-trained weight to train it more on far away drones images, with your yolo-drone.cfg file? or you think its better to start with yolo-tiny and darknet53 weights?

chuanenlin commented 5 years ago

@ngingihy Yes, you can continue training from the yolo-drone.weights in the repo.