/tianmouc_paper_code

Some demo codes for the paper "A Vision Sensor Chip with Complementary Pathways for Open-world Sensing"

Primary LanguagePythonMIT LicenseMIT

A vision chip with complementary pathways for open-world sensing

NEWS! The USB module of TianMouC is released.

NEWS! The Python library of TianMouC sensor is available in tianmouc/Tianmoucv_preview

The official version will be available at tianmouc/tianmocv

This repository contains the code of our 2024 Nature Cover paper. If you use our code or refer to this project, please cite it as

@article{yang2024vision,
  title={A vision chip with complementary pathways for open-world sensing},
  author={Yang, Zheyu and Wang, Taoyi and Lin, Yihan and Chen, Yuguo and Zeng, Hui and Pei, Jing and Wang, Jiazheng and Liu, Xue and Zhou, Yichun and Zhang, Jianqiang and others},
  journal={Nature},
  volume={629},
  number={8014},
  pages={1027--1033},
  year={2024},
  publisher={Nature Publishing Group UK London}
}

Arch of this project

├── data -> .../tianmouc_evaluation_data/
│   ├── ckpts
│   ├── demo_data
│   └── recon_data
├── datareader
├── demo
├── reconstruction
├── resources
└── tianmoucv

  • reconstruction : The reconstruction algorithm for TianMouC raw data

  • demo : the automotive driving perception algorithm

  • tianmoucv : some basic algorithm for TianMouC raw data

  • datareader : the raw data decoder and data reader

  • tianmouc_evaluation_data : Part of the labeled data for demo evaluation

    • ckpts : 3 Pytorch models used in this paper
    • demo_data : 5 samples used in Fig 4
    • recon_data : some clips for reconstruction

requirement

git clone  
conda create -n tianmouc python=3.10
sh install.sh

prepare Dataset

download the dataset in zenodo and decompress it.

use the soft link to create easy data access for this repo:

cd [N_pub_code]
ln -s [your dataset path] data

you can also download the checkpoints only in google driver

EASY START

The code is replicated using python with jupyter notebook

  • Fig. 4 with mAP evaluation
  /code/demo/Evaluation_complex.ipynb

fig4e

fig4e

the OF data need to be calculated using raw data

fig4e

You can change the key(name) of dataset to find more demos for Fig4 or automotive driving The labeled datasets are in /data/tianmouc_evaluation_data

  • anti-aliasing reconstruction
  /code/reconstruction/reconstruction.ipynb

fig4e