• Philip Meier's avatar
    Add tests for the PhotoTour dataset (#3486) · f637c63b
    Philip Meier authored
    * add tests for PhotoTour dataset
    
    * fix grayscale image generation
    
    * fix test_feature_types for a examples of a single feature
    
    * make image size variable instead of hard coding it
    
    * make dataset length variable instead of hard coding it
    
    * replace numpy with torch
    
    * fix typo
    f637c63b
test_datasets.py 52.2 KB