-
Notifications
You must be signed in to change notification settings - Fork 7.1k
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
Conversation
There was a problem hiding this 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
you also need to add |
… into add-crestereo-model
There was a problem hiding this 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.
There was a problem hiding this 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
There was a problem hiding this 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.
@TeodorPoncu Oh ya, also please resolve the conflict on the test files |
There was a problem hiding this 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!
There was a problem hiding this 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.
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]>
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.