BobLiu20 / YOLOv3_PyTorch

Full implementation of YOLOv3 in PyTorch
624 stars 191 forks source link

Implementing different backbone to this YOLOv3 #48

Open hilmandayo opened 5 years ago

hilmandayo commented 5 years ago

Hello.

First of all, thank you for this awesome repo!

I have one question. I want to change the backbone from darknet to something else. After inspected and tested the code, I found out that the things that I need to do are (roughly):

  1. Make sure that the new backbone net output three tensors from different layers in forward method
  2. While referring to nets/backbone/darknet.py, add some needed implementation details to the new backbone (like attributes and methods)
  3. Set the new backbone in params.py

Is this correct? Or is there any other things that I need to do or be careful of? Thanks.

mei123hao commented 3 years ago

Hello.

First of all, thank you for this awesome repo!

I have one question. I want to change the backbone from darknet to something else. After inspected and tested the code, I found out that the things that I need to do are (roughly):

  1. Make sure that the new backbone net output three tensors from different layers in forward method
  2. While referring to nets/backbone/darknet.py, add some needed implementation details to the new backbone (like attributes and methods)
  3. Set the new backbone in params.py

Is this correct? Or is there any other things that I need to do or be careful of? Thanks. Hi, I try to change the backbone same as you. However, the training loss reduce to 0.3 and stop falling, I try to test the saved weight file but got nothing. Do you solve your problems or have any adiveces?