• Yanghan Wang's avatar
    refactor create_fake_detection_data_loader · 312c6b62
    Yanghan Wang authored
    Summary:
    Pull Request resolved: https://github.com/facebookresearch/d2go/pull/199
    
    - `create_fake_detection_data_loader` currently doesn't take `cfg` as input, sometimes we need to test the augmentation that needs more complicated different cfg.
    - name is a bit bad, rename it to `create_detection_data_loader_on_toy_dataset`.
    - width/height were the resized size previously, we want to change it to the size of data source (image files) and use `cfg` to control resized size.
    
    Update V3:
    In V2 there're some test failures, the reason is that V2 is building data loader (via GeneralizedRCNN runner) using actual test config instead of default config before this diff + dataset name change. In V3 we uses the test's runner instead of default runner for the consistency. This reveals some real bugs that we didn't test before.
    
    Reviewed By: omkar-fb
    
    Differential Revision: D35238890
    
    fbshipit-source-id: 28a6037374e74f452f91b494bd455b38d3a48433
    312c6b62
test_data_loader.py 4.48 KB