| import os | |
| from dataset import DataLoaderTrain, DataLoaderVal, DataLoaderTest | |
| def get_training_data(rgb_dir, img_options, debug): | |
| assert os.path.exists(rgb_dir) | |
| return DataLoaderTrain(rgb_dir, img_options, None, debug) | |
| def get_validation_data(rgb_dir, debug=False): | |
| assert os.path.exists(rgb_dir) | |
| return DataLoaderVal(rgb_dir, None, debug) | |
| def get_test_data(rgb_dir, debug=False): | |
| assert os.path.exists(rgb_dir) | |
| return DataLoaderTest(rgb_dir, None, debug) | |