search

WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4
1.86k stars 585 forks source link

parameter 與 FLOPS 估算請教 #214

Closed peggy860629 closed 3 years ago

peggy860629 commented 3 years ago

作者你好: 想請教一般paper report 的 parameter size 與 FLOPS 的數字, 是否會包含yolo layer, NMS 這些呢?

非常感謝

WongKinYiu commented 3 years ago

一般有包含和沒包含看到的數字都一樣 對於動輒M等級的參數量和G等級的運算量 yolo layer和nms都影響不到看見的數字

輕量模型的話習慣上是把各個block分開分析

peggy860629 commented 3 years ago

非常感謝

indigohana commented 3 years ago

一般有包含和沒包含看到的數字都一樣 對於動輒M等級的參數量和G等級的運算量 yolo layer和nms都影響不到看見的數字

輕量模型的話習慣上是把各個block分開分析

你好，在Scaled-YOLOv4论文4.1章节的backbone中提到参数量的计算，请问9/4hwb^2这部分计算是怎么得到的呢？

WongKinYiu commented 3 years ago

Downsampling: 3x3 conv, stride 2. filter_szie x cin x cout x feature_map_size 3x3 x b x b x (h/2)x(w/2)

indigohana commented 3 years ago

Downsampling: 3x3 conv, stride 2. filter_szie x cin x cout x feature_map_size 3x3 x b x b x (h/2)x(w/2)

感谢！