├── LICENSE ├── README.md ├── callbacks ├── callbacks.py └── vis_callback.py ├── conf ├── dataset │ ├── base.yaml │ ├── gen1.yaml │ └── gen4.yaml ├── encoder_model │ ├── default.yaml │ └── encoder.yaml ├── experiment │ ├── gen1 │ │ ├── base.yaml │ │ ├── default.yaml │ │ ├── small.yaml │ │ └── tiny.yaml │ └── gen4 │ │ ├── base.yaml │ │ ├── default.yaml │ │ ├── small.yaml │ │ └── tiny.yaml ├── general.yaml ├── model │ ├── base.yaml │ ├── maxvit_yolox │ │ └── default.yaml │ └── rnndet.yaml ├── modifier.py ├── train.yaml └── val.yaml ├── data ├── genx_utils │ ├── collate.py │ ├── collate_from_pytorch.py │ ├── dataset_rnd.py │ ├── dataset_streaming.py │ ├── labels.py │ ├── sequence_base.py │ ├── sequence_for_streaming.py │ └── sequence_rnd.py └── utils │ ├── augmentor.py │ ├── representations.py │ ├── spatial.py │ ├── stream_concat_datapipe.py │ ├── stream_sharded_datapipe.py │ └── types.py ├── figs ├── 293152151-c68aa765-f58f-4434-8121-564dfd702b18.png ├── 293155252-0d45bd1f-7788-4054-ac08-474e4641fcf7.png └── 293155375-68793247-f2b3-4145-86a4-b08fc4671fcc.png ├── models ├── __init__.py ├── detection │ ├── __init_.py │ ├── recurrent_backbone │ │ ├── __init__.py │ │ ├── base.py │ │ └── maxvit_rnn.py │ ├── yolox │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── losses.py │ │ │ ├── network_blocks.py │ │ │ └── yolo_head.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── boxes.py │ │ │ └── compat.py │ └── yolox_extension │ │ └── models │ │ ├── __init__.py │ │ ├── build.py │ │ ├── detector.py │ │ └── yolo_pafpn.py ├── encoder │ ├── __init__.py │ ├── layers │ │ ├── __init__.py │ │ ├── mlp.py │ │ ├── patch_merging.py │ │ ├── sn.py │ │ ├── ssa.py │ │ └── temporal_extension.py │ └── spikingTransformerEncoder.py ├── layers │ ├── maxvit │ │ ├── __init__.py │ │ ├── layers │ │ │ ├── __init__.py │ │ │ ├── activations.py │ │ │ ├── activations_jit.py │ │ │ ├── activations_me.py │ │ │ ├── adaptive_avgmax_pool.py │ │ │ ├── attention_pool2d.py │ │ │ ├── blur_pool.py │ │ │ ├── bottleneck_attn.py │ │ │ ├── cbam.py │ │ │ ├── classifier.py │ │ │ ├── cond_conv2d.py │ │ │ ├── config.py │ │ │ ├── conv2d_same.py │ │ │ ├── conv_bn_act.py │ │ │ ├── create_act.py │ │ │ ├── create_attn.py │ │ │ ├── create_conv2d.py │ │ │ ├── create_norm.py │ │ │ ├── create_norm_act.py │ │ │ ├── drop.py │ │ │ ├── eca.py │ │ │ ├── evo_norm.py │ │ │ ├── fast_norm.py │ │ │ ├── filter_response_norm.py │ │ │ ├── gather_excite.py │ │ │ ├── global_context.py │ │ │ ├── halo_attn.py │ │ │ ├── helpers.py │ │ │ ├── inplace_abn.py │ │ │ ├── lambda_layer.py │ │ │ ├── linear.py │ │ │ ├── median_pool.py │ │ │ ├── mixed_conv2d.py │ │ │ ├── ml_decoder.py │ │ │ ├── mlp.py │ │ │ ├── non_local_attn.py │ │ │ ├── norm.py │ │ │ ├── norm_act.py │ │ │ ├── padding.py │ │ │ ├── patch_embed.py │ │ │ ├── pool2d_same.py │ │ │ ├── pos_embed.py │ │ │ ├── selective_kernel.py │ │ │ ├── separable_conv.py │ │ │ ├── space_to_depth.py │ │ │ ├── split_attn.py │ │ │ ├── split_batchnorm.py │ │ │ ├── squeeze_excite.py │ │ │ ├── std_conv.py │ │ │ ├── test_time_pool.py │ │ │ ├── trace_utils.py │ │ │ └── weight_init.py │ │ └── maxvit.py │ └── rnn.py └── spikingVit.py ├── modules ├── __init__.py ├── data │ └── genx.py ├── detection_module.py └── utils │ ├── detection.py │ └── fetch.py ├── requirements.txt ├── train.py ├── train_gen1.sh ├── train_gen4.sh ├── utils ├── Augmentation │ ├── __init__.py │ ├── autoaugment.py │ └── dataAugmentation.py ├── __init__.py ├── box_ops.py ├── ev_repr_to_img.py ├── evaluation │ ├── postprocess.py │ └── prophesee │ │ ├── __init__.py │ │ ├── evaluation.py │ │ ├── evaluator.py │ │ ├── io │ │ ├── __init__.py │ │ ├── box_filtering.py │ │ ├── box_loading.py │ │ ├── dat_events_tools.py │ │ ├── npy_events_tools.py │ │ └── psee_loader.py │ │ ├── metrics │ │ ├── __init__.py │ │ └── coco_eval.py │ │ └── visualize │ │ ├── __init__.py │ │ └── vis_utils.py ├── helpers.py ├── misc.py ├── padding.py ├── pos_embed.py └── timers.py ├── val_gen1.sh ├── val_gen4.sh └── val_or_test.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/README.md -------------------------------------------------------------------------------- /callbacks/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/callbacks/callbacks.py -------------------------------------------------------------------------------- /callbacks/vis_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/callbacks/vis_callback.py -------------------------------------------------------------------------------- /conf/dataset/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/dataset/base.yaml -------------------------------------------------------------------------------- /conf/dataset/gen1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/dataset/gen1.yaml -------------------------------------------------------------------------------- /conf/dataset/gen4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/dataset/gen4.yaml -------------------------------------------------------------------------------- /conf/encoder_model/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/encoder_model/default.yaml -------------------------------------------------------------------------------- /conf/encoder_model/encoder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/encoder_model/encoder.yaml -------------------------------------------------------------------------------- /conf/experiment/gen1/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/experiment/gen1/base.yaml -------------------------------------------------------------------------------- /conf/experiment/gen1/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/experiment/gen1/default.yaml -------------------------------------------------------------------------------- /conf/experiment/gen1/small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/experiment/gen1/small.yaml -------------------------------------------------------------------------------- /conf/experiment/gen1/tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/experiment/gen1/tiny.yaml -------------------------------------------------------------------------------- /conf/experiment/gen4/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/experiment/gen4/base.yaml -------------------------------------------------------------------------------- /conf/experiment/gen4/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/experiment/gen4/default.yaml -------------------------------------------------------------------------------- /conf/experiment/gen4/small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/experiment/gen4/small.yaml -------------------------------------------------------------------------------- /conf/experiment/gen4/tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/experiment/gen4/tiny.yaml -------------------------------------------------------------------------------- /conf/general.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/general.yaml -------------------------------------------------------------------------------- /conf/model/base.yaml: -------------------------------------------------------------------------------- 1 | name: ??? -------------------------------------------------------------------------------- /conf/model/maxvit_yolox/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/model/maxvit_yolox/default.yaml -------------------------------------------------------------------------------- /conf/model/rnndet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/model/rnndet.yaml -------------------------------------------------------------------------------- /conf/modifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/modifier.py -------------------------------------------------------------------------------- /conf/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/train.yaml -------------------------------------------------------------------------------- /conf/val.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/conf/val.yaml -------------------------------------------------------------------------------- /data/genx_utils/collate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/genx_utils/collate.py -------------------------------------------------------------------------------- /data/genx_utils/collate_from_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/genx_utils/collate_from_pytorch.py -------------------------------------------------------------------------------- /data/genx_utils/dataset_rnd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/genx_utils/dataset_rnd.py -------------------------------------------------------------------------------- /data/genx_utils/dataset_streaming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/genx_utils/dataset_streaming.py -------------------------------------------------------------------------------- /data/genx_utils/labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/genx_utils/labels.py -------------------------------------------------------------------------------- /data/genx_utils/sequence_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/genx_utils/sequence_base.py -------------------------------------------------------------------------------- /data/genx_utils/sequence_for_streaming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/genx_utils/sequence_for_streaming.py -------------------------------------------------------------------------------- /data/genx_utils/sequence_rnd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/genx_utils/sequence_rnd.py -------------------------------------------------------------------------------- /data/utils/augmentor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/utils/augmentor.py -------------------------------------------------------------------------------- /data/utils/representations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/utils/representations.py -------------------------------------------------------------------------------- /data/utils/spatial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/utils/spatial.py -------------------------------------------------------------------------------- /data/utils/stream_concat_datapipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/utils/stream_concat_datapipe.py -------------------------------------------------------------------------------- /data/utils/stream_sharded_datapipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/utils/stream_sharded_datapipe.py -------------------------------------------------------------------------------- /data/utils/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/data/utils/types.py -------------------------------------------------------------------------------- /figs/293152151-c68aa765-f58f-4434-8121-564dfd702b18.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/figs/293152151-c68aa765-f58f-4434-8121-564dfd702b18.png -------------------------------------------------------------------------------- /figs/293155252-0d45bd1f-7788-4054-ac08-474e4641fcf7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/figs/293155252-0d45bd1f-7788-4054-ac08-474e4641fcf7.png -------------------------------------------------------------------------------- /figs/293155375-68793247-f2b3-4145-86a4-b08fc4671fcc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/figs/293155375-68793247-f2b3-4145-86a4-b08fc4671fcc.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/detection/__init_.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/detection/recurrent_backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/detection/recurrent_backbone/__init__.py -------------------------------------------------------------------------------- /models/detection/recurrent_backbone/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/detection/recurrent_backbone/base.py -------------------------------------------------------------------------------- /models/detection/recurrent_backbone/maxvit_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/detection/recurrent_backbone/maxvit_rnn.py -------------------------------------------------------------------------------- /models/detection/yolox/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/detection/yolox/models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/detection/yolox/models/losses.py -------------------------------------------------------------------------------- /models/detection/yolox/models/network_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/detection/yolox/models/network_blocks.py -------------------------------------------------------------------------------- /models/detection/yolox/models/yolo_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/detection/yolox/models/yolo_head.py -------------------------------------------------------------------------------- /models/detection/yolox/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/detection/yolox/utils/__init__.py -------------------------------------------------------------------------------- /models/detection/yolox/utils/boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/detection/yolox/utils/boxes.py -------------------------------------------------------------------------------- /models/detection/yolox/utils/compat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/detection/yolox/utils/compat.py -------------------------------------------------------------------------------- /models/detection/yolox_extension/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/detection/yolox_extension/models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/detection/yolox_extension/models/build.py -------------------------------------------------------------------------------- /models/detection/yolox_extension/models/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/detection/yolox_extension/models/detector.py -------------------------------------------------------------------------------- /models/detection/yolox_extension/models/yolo_pafpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/detection/yolox_extension/models/yolo_pafpn.py -------------------------------------------------------------------------------- /models/encoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/encoder/__init__.py -------------------------------------------------------------------------------- /models/encoder/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/encoder/layers/__init__.py -------------------------------------------------------------------------------- /models/encoder/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/encoder/layers/mlp.py -------------------------------------------------------------------------------- /models/encoder/layers/patch_merging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/encoder/layers/patch_merging.py -------------------------------------------------------------------------------- /models/encoder/layers/sn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/encoder/layers/sn.py -------------------------------------------------------------------------------- /models/encoder/layers/ssa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/encoder/layers/ssa.py -------------------------------------------------------------------------------- /models/encoder/layers/temporal_extension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/encoder/layers/temporal_extension.py -------------------------------------------------------------------------------- /models/encoder/spikingTransformerEncoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/encoder/spikingTransformerEncoder.py -------------------------------------------------------------------------------- /models/layers/maxvit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/layers/maxvit/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/__init__.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/activations.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/activations_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/activations_jit.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/activations_me.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/activations_me.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/adaptive_avgmax_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/adaptive_avgmax_pool.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/attention_pool2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/attention_pool2d.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/blur_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/blur_pool.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/bottleneck_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/bottleneck_attn.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/cbam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/cbam.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/classifier.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/cond_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/cond_conv2d.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/config.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/conv2d_same.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/conv2d_same.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/conv_bn_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/conv_bn_act.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/create_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/create_act.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/create_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/create_attn.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/create_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/create_conv2d.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/create_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/create_norm.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/create_norm_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/create_norm_act.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/drop.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/eca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/eca.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/evo_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/evo_norm.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/fast_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/fast_norm.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/filter_response_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/filter_response_norm.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/gather_excite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/gather_excite.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/global_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/global_context.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/halo_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/halo_attn.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/helpers.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/inplace_abn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/inplace_abn.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/lambda_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/lambda_layer.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/linear.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/median_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/median_pool.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/mixed_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/mixed_conv2d.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/ml_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/ml_decoder.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/mlp.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/non_local_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/non_local_attn.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/norm.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/norm_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/norm_act.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/padding.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/patch_embed.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/pool2d_same.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/pool2d_same.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/pos_embed.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/selective_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/selective_kernel.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/separable_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/separable_conv.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/space_to_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/space_to_depth.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/split_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/split_attn.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/split_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/split_batchnorm.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/squeeze_excite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/squeeze_excite.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/std_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/std_conv.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/test_time_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/test_time_pool.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/trace_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/trace_utils.py -------------------------------------------------------------------------------- /models/layers/maxvit/layers/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/layers/weight_init.py -------------------------------------------------------------------------------- /models/layers/maxvit/maxvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/maxvit/maxvit.py -------------------------------------------------------------------------------- /models/layers/rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/layers/rnn.py -------------------------------------------------------------------------------- /models/spikingVit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/models/spikingVit.py -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/data/genx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/modules/data/genx.py -------------------------------------------------------------------------------- /modules/detection_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/modules/detection_module.py -------------------------------------------------------------------------------- /modules/utils/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/modules/utils/detection.py -------------------------------------------------------------------------------- /modules/utils/fetch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/modules/utils/fetch.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/train.py -------------------------------------------------------------------------------- /train_gen1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/train_gen1.sh -------------------------------------------------------------------------------- /train_gen4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/train_gen4.sh -------------------------------------------------------------------------------- /utils/Augmentation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/Augmentation/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/Augmentation/autoaugment.py -------------------------------------------------------------------------------- /utils/Augmentation/dataAugmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/Augmentation/dataAugmentation.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/box_ops.py -------------------------------------------------------------------------------- /utils/ev_repr_to_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/ev_repr_to_img.py -------------------------------------------------------------------------------- /utils/evaluation/postprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/evaluation/postprocess.py -------------------------------------------------------------------------------- /utils/evaluation/prophesee/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/evaluation/prophesee/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/evaluation/prophesee/evaluation.py -------------------------------------------------------------------------------- /utils/evaluation/prophesee/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/evaluation/prophesee/evaluator.py -------------------------------------------------------------------------------- /utils/evaluation/prophesee/io/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/evaluation/prophesee/io/box_filtering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/evaluation/prophesee/io/box_filtering.py -------------------------------------------------------------------------------- /utils/evaluation/prophesee/io/box_loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/evaluation/prophesee/io/box_loading.py -------------------------------------------------------------------------------- /utils/evaluation/prophesee/io/dat_events_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/evaluation/prophesee/io/dat_events_tools.py -------------------------------------------------------------------------------- /utils/evaluation/prophesee/io/npy_events_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/evaluation/prophesee/io/npy_events_tools.py -------------------------------------------------------------------------------- /utils/evaluation/prophesee/io/psee_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/evaluation/prophesee/io/psee_loader.py -------------------------------------------------------------------------------- /utils/evaluation/prophesee/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/evaluation/prophesee/metrics/coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/evaluation/prophesee/metrics/coco_eval.py -------------------------------------------------------------------------------- /utils/evaluation/prophesee/visualize/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/evaluation/prophesee/visualize/vis_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/evaluation/prophesee/visualize/vis_utils.py -------------------------------------------------------------------------------- /utils/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/helpers.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/padding.py -------------------------------------------------------------------------------- /utils/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/pos_embed.py -------------------------------------------------------------------------------- /utils/timers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/utils/timers.py -------------------------------------------------------------------------------- /val_gen1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/val_gen1.sh -------------------------------------------------------------------------------- /val_gen4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/val_gen4.sh -------------------------------------------------------------------------------- /val_or_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chrazqee/SpikingVit/HEAD/val_or_test.py --------------------------------------------------------------------------------