I wrote a tensorflow version of Yolo based on the pytorch version of Yolo, but it seems to have many problems, including
1: I feel that my custom layer layer does not handle well. They do not seem to pass any variables?
2: When I start training, I will instantly run out of 2G video memory, but I don’t know what the problem is. I will batch_ Adjusting the size to 8 does not solve this problem. It seems that my model is wrong
3: When previewing my model, there is no output shape, and some params are also none. What is the problem? (as shown below)
I would be grateful if someone could give me some advice
and This is all my code, and the comments in it are all in Chinese, because almost all my comments indicate my inner thoughts at that time