BobLiu20/YOLOv3_PyTorch

Implementing different backbone to this YOLOv3

Opened this issue · 1 comments

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.

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?