Skip to content

Attention-Aware Feature Aggregation for Real-time Stereo Matching on Edge Devices (ACCV, 2020)

License

Notifications You must be signed in to change notification settings

JiaRenChang/RealtimeStereo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Attention-Aware Feature Aggregation for Real-time Stereo Matching on Edge Devices

This repository contains the code (in PyTorch) for "Attention-Aware Feature Aggregation for Real-time Stereo Matching on Edge Devices" paper (ACCV 2020) by Jia-Ren Chang, Pei-Chun Chang and Yong-Sheng Chen.

The codes mainly bring from PSMNet.

Citation

@InProceedings{Chang_2020_ACCV,
    author    = {Chang, Jia-Ren and Chang, Pei-Chun and Chen, Yong-Sheng},
    title     = {Attention-Aware Feature Aggregation for Real-time Stereo Matching on Edge Devices},
    booktitle = {Proceedings of the Asian Conference on Computer Vision (ACCV)},
    month     = {November},
    year      = {2020}
}

Train

As an example, use the following command to train a RTStereo on Scene Flow

python main.py --maxdisp 192 \
               --model RTStereoNet \
               --datapath (your scene flow data folder)\
               --epochs 10 \
               --loadmodel (optional)\
               --savemodel (path for saving model)

Pretrained Model

KITTI 2015 Pretrained Model Google Drive

About

Attention-Aware Feature Aggregation for Real-time Stereo Matching on Edge Devices (ACCV, 2020)

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published