Skip to content

rhwxmx/SECNet_ICML

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

We are collaborating with Beihang University on FPGA hardware development. The relevant code will be released upon completion of this work. SECNet's architecture

sample1 sample2 sample3 sample4

Installation

conda create -n secnet python=3.8
conda activate secnet
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia
conda install h5py,tqdm,scikit-learn,tensorboard
pip install spikingjelly
optional: install the cuda kernel: https://github.com/erikwijmans/Pointnet2_PyTorch

Usage

  1. Prepare the data:

     cd dataprocess
     python generate_xxx.py
    
  2. Put the train.h5 and test.h5 to ./data/xxx/:

  3. Modify the num_class, data_path, log_name and others.

  4. Run the train script:

     python train_frequency.py
    

Configuration

Dataset PointNumber Dimension Group Accuracy
N-MNIST 4096 32 512 0.997
N-CARS 8192 64 512 0.947
CIFAR10-DVS 10240 64 2048 0.757
N-Caltech101 8192 64 2048 0.824
ASL-DVS 4096 32 1024 0.999
DVSGesture 1024 64 512 0.989
DailyDVS 8192 64 1024 0.9965
UCF101-DVS 8192 64 1024 0.916
THU-E-ACT 8192 64 1024 0.9725
DHP19 4096 64 512 6.11/69.89

Citation

If you find our work useful in your research, please consider citing:

    @inproceedings{
    anonymous2026scalable,
    title={Scalable Event Cloud Network for Event-based Classification},
    author={Anonymous},
    booktitle={Forty-third International Conference on Machine Learning},
    year={2026},
    url={https://openreview.net/forum?id=yAAUcDLYMR}
    }

and this paper is related to our previous three works EventMamba, TTPOINT and PEPNet:

@article{ren2024rethinking,
title={Rethinking Efficient and Effective Point-based Networks for Event Camera Classification and Regression: EventMamba},
author={Ren, Hongwei and Zhou, Yue and Zhu, Jiadong and Fu, Haotian and Huang, Yulong and Lin, Xiaopeng and Fang, Yuetong and Ma, Fei and Yu, Hao and Cheng, Bojun},
journal={arXiv preprint arXiv:2405.06116},
year={2024}
}
@inproceedings{ren2023ttpoint,
title={Ttpoint: A tensorized point cloud network for lightweight action recognition with event cameras},
author={Ren, Hongwei and Zhou, Yue and Fu, Haotian and Huang, Yulong and Xu, Renjing and Cheng, Bojun},
booktitle={Proceedings of the 31st ACM International Conference on Multimedia},
pages={8026--8034},
year={2023}
}
@inproceedings{ren2024simple,
title={A Simple and Effective Point-based Network for Event Camera 6-DOFs Pose Relocalization},
author={Ren, Hongwei and Zhu, Jiadong and Zhou, Yue and Fu, Haotian and Huang, Yulong and Cheng, Bojun},
booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
pages={18112--18121},
year={2024}
}    

Acknowledgment

Thanks to the previous works, PointNet, PointNet++, PointMLP, EventPointPose and STNet.

About

Scalable Event Cloud Network for Event-based Classification

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors