I found some problems

[zhangyahui520]

Hello Jay Meng, I've looked over your code and I think your work is very good.Your work has been a great help to me.But I find that there is something wrong with your network structure, which I think is the reason why your work is not working well at present.I am sending you a summary of my work, hoping it will be of some help to you.Looking forward to your more excellent content.

The focus operation is missing at the beginning of your network, which is similar to a down sampling operation.Specific I recommend you check the details of the article: https://zhuanlan.zhihu.com/p/164627427, this article analysis the very detailed, I hope you can made great achievements。

[avBuffer]

Hey Yahui, Thanks! I‘ll study the article and try to do better.

[dolortaste]

try tf.space_to_depth