search

ken2576 / vision-nerf

Official PyTorch Implementation of paper "Vision Transformer for NeRF-Based View Synthesis from a Single Input Image", WACV 2023.
MIT License
107 stars 12 forks source link

Excuse me，what part of the code corresponds to local feature extraction ? #12

Closed Yancy-lv closed 1 year ago

Yancy-lv commented 1 year ago

Excuse me, what part of the code corresponds to local feature extraction?

ken2576 commented 1 year ago

IIRC, It's https://github.com/ken2576/vision-nerf/blob/main/network/vit.py#L468-L488 and https://github.com/ken2576/vision-nerf/blob/main/network/vit.py#L548-L549