PyTorch-Spiking-YOLOv3 icon indicating copy to clipboard operation
PyTorch-Spiking-YOLOv3 copied to clipboard

Training on VOC dataset

Open polarcrow opened this issue 4 years ago • 1 comments

Hello,

I want to train the Yolov3-tiny-ours model on the VOC dataset but didn't find the code and dataset download script. How did you train the model on the VOC dataset ?

Also, do you plan to integrate channel-wise parameters normalization and leaky ReLu spiking activations like exposed in the Spiking-YOLO paper as I didn't find references to them in your code ?

Thanks in advance.

polarcrow avatar Mar 09 '21 13:03 polarcrow

Please download the VOC dataset by yourself. This code has integrated channel-wise parameters normalization. You can use it with - -channel-wise. This code hasn’t included leaky ReLU spiking activations because we only use positive spike(+1) not negative spike(-1). Leaky ReLU uses negative spike(-1) which we will code for it in the future.

cwq159 avatar Mar 10 '21 03:03 cwq159