Roman Shapovalov 09a99f2e6d Support limiting num sequences per category. ...

Summary:
Adds stratified sampling of sequences within categories applied after category / sequence filters but before the num sequence limit.
It respects the insertion order into the sequence_annots table, i.e. takes top N sequences within each category.

Reviewed By: bottler

Differential Revision: D46724002

fbshipit-source-id: 597cb2a795c3f3bc07f838fc51b4e95a4f981ad3

2023-06-14 07:12:02 -07:00

..

data

SQL Index Dataset

2023-04-25 09:56:15 -07:00

models

Fix inversion between fine and coarse implicit_functions

2023-05-26 08:33:54 -07:00

__init__.py

implicitron v0 (#1133)

2022-03-21 13:20:10 -07:00

common_resources.py

resources fix

2023-05-19 06:41:33 -07:00

test_batch_sampler.py

Extract BlobLoader class from JsonIndexDataset and moving crop_by_bbox to FrameData

2023-04-04 07:17:43 -07:00

test_bbox.py

Extract BlobLoader class from JsonIndexDataset and moving crop_by_bbox to FrameData

2023-04-04 07:17:43 -07:00

test_build.py

Add MeshRasterizerOpenGL

2022-07-22 15:52:50 -07:00

test_circle_fitting.py

Fix returning a proper rotation in levelling; supporting batches and default centroid

2022-09-29 11:56:14 -07:00

test_co3d_sql.py

SQL Index Dataset

2023-04-25 09:56:15 -07:00

test_config_use.py

Move load_stats to TrainingLoop

2022-08-02 15:40:53 -07:00

test_config.py

Hard population of registry system with pre_expand

2023-03-31 07:44:38 -07:00

test_data_cow.py

Extract BlobLoader class from JsonIndexDataset and moving crop_by_bbox to FrameData

2023-04-04 07:17:43 -07:00

test_data_json_index.py

Subsets in dataset iterators

2023-01-26 03:00:46 -08:00

test_data_llff.py

Mods and bugfixes for LLFF and Blender repros

2022-08-09 15:04:44 -07:00

test_data_source.py

Fix test_data_source in OSS.

2023-05-05 02:05:50 -07:00

test_dataset_visualize.py

Optional ground-truth depth maps in visualiser

2023-04-18 07:00:17 -07:00

test_eval_cameras.py

move targets

2022-05-25 06:16:03 -07:00

test_eval_demo.py

PathManagerFactory

2022-06-10 12:22:46 -07:00

test_evaluation.py

Extract BlobLoader class from JsonIndexDataset and moving crop_by_bbox to FrameData

2023-04-04 07:17:43 -07:00

test_extending_orm_types.py

More tests for SQL Dataset

2023-05-04 03:32:27 -07:00

test_forward_pass.py

remove stray "generic_model_args" references

2022-08-18 07:18:55 -07:00

test_frame_data_builder.py

Refactor: FrameDataBuilder is more extensible.

2023-05-17 10:38:34 -07:00

test_json_index_dataset_provider_v2.py

Extract BlobLoader class from JsonIndexDataset and moving crop_by_bbox to FrameData

2023-04-04 07:17:43 -07:00

test_model_visualize.py

Bugfixes in render_flyaround

2022-09-06 05:45:31 -07:00

test_orm_types.py

Serialising dynamic arrays in SQL; read-only SQLite connection in SQL Dataset

2023-05-22 02:24:49 -07:00

test_pointcloud_utils.py

doc rgbd point cloud

2023-01-24 15:26:52 -08:00

test_ray_point_refiner.py

ImplicitronRayBundle

2022-10-03 08:36:47 -07:00

test_sql_dataset.py

Support limiting num sequences per category.

2023-06-14 07:12:02 -07:00

test_srn.py

ImplicitronRayBundle

2022-10-03 08:36:47 -07:00

test_types.py

Fix: typo in dict processing

2022-06-28 16:11:49 -07:00

test_viewsampling.py

ViewPooler class

2022-05-12 12:50:03 -07:00

test_voxel_grid_implicit_function.py

voxel_grid_implicit_function scaffold fixes

2022-11-03 05:46:31 -07:00

test_voxel_grids.py

different learning rate for different parts

2022-10-18 15:58:18 -07:00