Astronet loader - #258
Conversation
|
|
||
| TEST_DATA_ROOT = Path(__file__).resolve().parent.parent / "data" | ||
| TEST_POINT3D_IDS = [7, 2560, 3063, 3252, 3424, 3534, 3653, 3786, 3920, 4062, 4420, 4698, 3311, 3963, 325] | ||
| TEST_SFMTRACKS_INDICES = [0, 100, 200, 300, 400, 500, 600, 700, 800, 900, 1000, 1100, 1200, 1300, 1400] |
There was a problem hiding this comment.
could you add an inline comment explaining where these come from and how they were selected?
There was a problem hiding this comment.
Explanation added.
There was a problem hiding this comment.
If these have a 1:1 correspondences, it might be better to use a list of tuples instead of 2 lists. Ive found that to cause fewer bugs.
There was a problem hiding this comment.
They're used individually as well, so I thought it would be clearer to split them up.
|
Please provide a PR description. I'll just let John and Ayush review this, as they are familiar with the loaders. :) |
|
Is the figure you posted from the deep front end? |
Yes it is |
cool, indeed lots of room for improvement :-) |
|
Thanks for all the updates, Travis. Can we put the Astronet test data under an astronet subdir? instead of |
|
Looks quite good. Thanks Travis. @ayushbaid could you take another look when you get a chance since you are the most familiar with the |
|
|
||
| from gtsfm.common.image import Image | ||
| from gtsfm.loader.astronet_loader import AstroNetLoader | ||
| from thirdparty.colmap.scripts.python.read_write_model import read_model |
There was a problem hiding this comment.
same note here about importing an individual function vs. Google style
| logger = logger_utils.get_logger() | ||
|
|
||
|
|
||
| class AstroNetLoader(LoaderBase): |
There was a problem hiding this comment.
Should the class be called AstronetLoader, considering the context here? The file is also called astronet_loader.
There was a problem hiding this comment.
My dataset is called AstroNet, so yeah I think it should be AstronetLoader instead.
There was a problem hiding this comment.
Akshay did you mean disambiguating from a CNN architecture/network name here?
There was a problem hiding this comment.
Yeah I just meant using a small 'n' instead of 'N' because Astronet is one thing as far as the loader is concerned. Its resolved now.
|
|
||
| TEST_DATA_ROOT = Path(__file__).resolve().parent.parent / "data" | ||
| TEST_POINT3D_IDS = [7, 2560, 3063, 3252, 3424, 3534, 3653, 3786, 3920, 4062, 4420, 4698, 3311, 3963, 325] | ||
| TEST_SFMTRACKS_INDICES = [0, 100, 200, 300, 400, 500, 600, 700, 800, 900, 1000, 1100, 1200, 1300, 1400] |
There was a problem hiding this comment.
If these have a 1:1 correspondences, it might be better to use a list of tuples instead of 2 lists. Ive found that to cause fewer bugs.
| # assert self.loader._gt_scene_trimesh is not None | ||
| # assert self.loader._gt_scene_trimesh.vertices;shape[0] == 5002 |
There was a problem hiding this comment.
remove if unnecessary?
There was a problem hiding this comment.
I will be uncommenting this (and some other tests) in the next PR when I add trimesh. It would be useful to be able to keep them here so I don't have to rewrite them.

No description provided.