Skip to content

add crestereo implementation #6310

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 23 commits into from
Sep 15, 2022
Merged

add crestereo implementation #6310

merged 23 commits into from
Sep 15, 2022

Conversation

TeodorPoncu
Copy link
Contributor

This is a draft version for the CREStereo model implementation.

It's over-commented for easier readability for those that are unfamiliar with the paper.

It also makes use of ModuleDicts and Dicts to process / access feature pyramid level instead of ModuleLists and Lists for readability purposes.

@TeodorPoncu TeodorPoncu changed the title crestereo draft implementation [WIP] crestereo draft implementation Jul 25, 2022
Copy link
Contributor

@jdsgomes jdsgomes left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

First pass, looks good in general so only left a few comments

@jdsgomes
Copy link
Contributor

you also need to add from .crestereo import * to torchvision/prototype/models/depth/stereo/__init__.py

@jdsgomes jdsgomes changed the title [WIP] crestereo draft implementation add crestereo implementation Sep 8, 2022
Copy link
Contributor

@YosuaMichael YosuaMichael left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice work @TeodorPoncu ! Overall it looks pretty good. I have several comments that mostly NIT.

Copy link
Contributor

@YosuaMichael YosuaMichael left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hi @TeodorPoncu, overall looks good, I just have a few more comments

Copy link
Contributor

@YosuaMichael YosuaMichael left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hi @TeodorPoncu, I think this is very close to a merge. Just have few things left that you forgot to update and NITs.

@YosuaMichael
Copy link
Contributor

@TeodorPoncu Oh ya, also please resolve the conflict on the test files

Copy link
Contributor

@YosuaMichael YosuaMichael left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for your hard work @TeodorPoncu !
The PR looks good to me!

@jdsgomes could you give a one more pass of review as well? Thanks!

Copy link
Contributor

@jdsgomes jdsgomes left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thank you for addressing all the comments. I think the weights can be added in a separate PR.

@jdsgomes jdsgomes merged commit 1c3eedc into main Sep 15, 2022
facebook-github-bot pushed a commit that referenced this pull request Sep 15, 2022
Summary:
* crestereo draft implementation

* minor model fixes. positional embedding changes.

* aligned base configuration with paper

* Adressing comments

* Broke down Adaptive Correlation Layer. Adressed some other commets.

* adressed some nits

* changed search size, added output channels to model attrs

* changed weights naming

* changed from iterations to num_iters

* removed _make_coords, adressed comments

* fixed jit test

* config nit

* Changed device arg to str

Reviewed By: jdsgomes

Differential Revision: D39543279

fbshipit-source-id: c6101958588eb43201f92ff4f687bd32cbbcbbd1

Co-authored-by: Joao Gomes <[email protected]>
Co-authored-by: YosuaMichael <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants