search

BICLab / SpikeYOLO

Offical implementation of "Integer-Valued Training and Spike-Driven Inference Spiking Neural Network for High-performance and Energy-efficient Object Detection" (ECCV2024 Best Paper Candidate / Oral)
https://arxiv.org/abs/2407.20708
GNU Affero General Public License v3.0
111 stars 6 forks source link

参数量计算方面 #8

Closed jiaduob closed 1 month ago

jiaduob commented 1 month ago

你好，感谢你们的工作，我在自己的数据集上成功复现SpikeYOLO并取得了不错的效果。但是同时我有一些疑问希望得到解答。

1.参数量计算方面

在paper中是这样的（配置文件中有5种scale分别是nslmx，这里是少了哪一种？）： SpikeYOLO参数量-论文

在项目中的配置文件snn_yolov8.yaml中是这样描述的： SpikeYOLO参数量-配置文件

ultralytics的自带的功能进行scale为s的计算： SpikeYOLO参数量-计算

请问哪一种是准确的？

XinhaoLuo666 commented 1 month ago

您好，论文中的参数量和ultralytics的自带计算得到的结果是正确的，项目配置文件中的参数量是原版yolov8的参数量，我们将在后续版本中删除。我们没有训练scale为x的模型。

jiaduob commented 1 month ago

非常感谢您的回复