bubbliiiing / mobilenet-yolov4-pytorch

这是一个mobilenet-yolov4的库,把yolov4主干网络修改成了mobilenet,修改了Panet的卷积组成,使参数量大幅度缩小。
MIT License
366 stars 104 forks source link

如何获得自己修改后的网络参数数量!? #6

Closed ghost closed 3 years ago

ghost commented 3 years ago

如何获得像大神这样的参数量

# mobilenetv1-yolov4 40,952,893
# mobilenetv2-yolov4 39,062,013
# mobilenetv3-yolov4 39,989,933

# 修改了panet的mobilenetv1-yolov4 12,692,029
# 修改了panet的mobilenetv2-yolov4 10,801,149
# 修改了panet的mobilenetv3-yolov4 11,729,069

新年快乐!!!!!

bubbliiiing commented 3 years ago

你不是在test.py看到的吗

bubbliiiing commented 3 years ago

新年快乐

ghost commented 3 years ago

感谢!