bubbliiiing / mobilenet-yolov4-pytorch

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

观看网络结构的问题 #26

Open yanyanxinshi opened 2 years ago

yanyanxinshi commented 2 years ago

大佬您好,在summary函数中查看网络结构时,一直会有错误,显示为未解析的引用 'summary'。 image

yanyanxinshi commented 2 years ago

已解决,未安装torchsummary包导致的

bubbliiiing commented 2 years ago

好的~