Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

training slightly modified yolov4-tiny with imagenet #6352

Open
huseyinuri opened this issue Jul 27, 2020 · 4 comments
Open

training slightly modified yolov4-tiny with imagenet #6352

huseyinuri opened this issue Jul 27, 2020 · 4 comments

Comments

@huseyinuri
Copy link

huseyinuri commented Jul 27, 2020

I am planning to use yolov4-tiny for an embedded device (rockchip rk3399pro) but groups and group_id are not supported parameters for model conversion toolkit. If i understood correctly, route layer slices the input feature map into 2 parts (groups = 2) and forwards the second part (group_id = 1) to the next convolutional layer. It seems that replacing route layers (not concatenating ones) with 1x1 convolution layers are supposed to work. However, training fails and loss becomes nan after a while with modified cfg with yolov4-tiny.conv.29.

  1. Original code piece
    image

    Route layer replaced with 1x1 convolution
    image

My questions

  1. Is it ok to replace some route layers with 1x1 convolutions ?
  2. Should i retrain yolov4-tiny backbone with imagenet from scratch first ? If yes, how ?
  3. What are the alternative backbones for yolov4-tiny without having groups params ?
@huseyinuri
Copy link
Author

Shufflenet backbone uses channel_slice for splitting feature map. Is it ok to replace some route layers with channel_slice? : #3750

@asmaamirkhan
Copy link

asmaamirkhan commented Jul 28, 2020

Is it solved? 👀
I have the same problem 🙄

@kadirbeytorun
Copy link

I asked a similar question and did not receive a proper answer.
@AlexeyAB @WongKinYiu

@diPDew
Copy link

diPDew commented Dec 26, 2020

Same question here. Any update so far?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants