zhen8838 / K210-yolo3

A Tensorflow implementation of keras-YOLOv3. Now I am ported to k210
MIT License
64 stars 9 forks source link

Output shape #5

Closed krishnak closed 4 years ago

krishnak commented 4 years ago

@zhen8838

In this code

What is the shape of output 0 and 1 which comes from KPU

zhen8838 commented 4 years ago

c array. len=C x H x W

krishnak commented 4 years ago

I mean is that a single dimension array? or can it be a 3 dimensional array?

For ex: Keras output is 3x7x10 is this have to be reshaped to 1x210 in KPU or can the KPU also output 3x7x10

krishnak commented 4 years ago

I think the confusion arose from the difference between your code

where the output is of shape 10x7x75

however in Kendryte repository this is mentioned as 10x7x125

You are using V3 anchors, Kendryte is using V2 anchors

zhen8838 commented 4 years ago
krishnak commented 4 years ago

yes got it thanks