https://github.com/Dignity-ghost/PyTorch-Spiking-YOLOv3

https://github.com/Dignity-ghost/PyTorch-Spiking-YOLOv3

Last synced: 12 days ago
JSON representation

• Host: GitHub
• URL: https://github.com/Dignity-ghost/PyTorch-Spiking-YOLOv3
• Owner: Dignity-ghost
• License: gpl-3.0
• Created: 2021-09-13T15:29:40.000Z (almost 3 years ago)
• Default Branch: ultralytics
• Last Pushed: 2021-09-15T05:44:33.000Z (almost 3 years ago)
• Last Synced: 2024-02-29T09:34:58.428Z (4 months ago)
• Language: Python
• Size: 4.82 MB
• Stars: 4
• Watchers: 1
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Lists

• awesome-yolo-object-detection - Dignity-ghost/PyTorch-Spiking-YOLOv3 - ghost/PyTorch-Spiking-YOLOv3?style=social"/> : A modified repository based on [Spiking-YOLOv3](https://github.com/cwq159/PyTorch-Spiking-YOLOv3) and [YOLOv3](https://pjreddie.com/darknet/yolo), which makes it suitable for VOC-dataset and YOLOv2. (Applications)

README

        
# PyTorch-Spiking-YOLOv3

A modified repository based on PyTorch-Spiking-YOLOv3([cwq159/PyTorch-Spiking-YOLOv3](https://github.com/cwq159/PyTorch-Spiking-YOLOv3)) and YOLOv3([ultralytics/yolov3](https://github.com/ultralytics/yolov3)), which makes it suitable for VOC-dataset and YOLOv2. There is no extra contribution, and thanks for the two authors above. It is feasible for CNN training and testing and SNN testing.

For yolov2 or yolov3 training, it is useful to refer https://github.com/ultralytics/yolov3/issues/1570 and get original code as below:

``` bash

$ git clone https://github.com/ultralytics/yolov3 -b archive  # archive branch

```

## Dataset

VOC dataset should be the same state as PyTorch-Spiking-YOLOv3, such as:

/parent

  /dataset/VOC

  /PyTorch-Spiking-YOLOv3

Please get VOC.tar in dataset and upzip them.

Use voc_label.py in dataset (from https://pjreddie.com/media/files/voc_label.py) to generate labels.

Use rebuild_voc.py in VOC to make a new collection of VOCdatset in folder VOC.

## Introduction

For spiking implementation, some operators in YOLOv2-Tiny have been converted equivalently. Please refer to yolov3-tiny-ours(\*).cfg in /cfg for details.

### Conversion of some operators

+ 'maxpool(stride=2)'->'convolutional(stride=2)'

+ 'maxpool(stride=1)'->'none'

+ 'leaky_relu'->'relu'

+ 'batch_normalization'->'fuse_conv_and_bn'

## Usage

Please refer to [ultralytics/yolov3](https://github.com/ultralytics/yolov3) for the basic usage for training, evaluation and inference. 

### Train

```

$ python3 train.py

```

### Test

```

$ python3 test.py

```

### Detect

```

$ python3 detect.py

```

### Transform

```

$ python3 ann_to_snn.py

```

For higher accuracy(mAP), you can try to adjust some hyperparameters.

*Trick: the larger timesteps, the higher accuracy.*