def get_loader(mode):
    path = '/home/ray/Data/KITTI/testing' if mode == 'test' else '/home/ray/Data/KITTI/training'
    source = KITTISource(path, max_frames=10000)
    return TripletDetectionLoader(source,
                                  crop_size=[255, 255],
                                  anchor_size=[127, 127],
                                  obj_types=['Car'],
                                  mode=mode)
def get_loader(mode):
    path = '/home/ray/Data/KITTI/testing' if mode == 'test' else '/home/ray/Data/KITTI/training'
    source = KITTISource(path, max_frames=10000)
    loader_params = {
        'crop_size': [255, 255],
        'anchor_size': [127, 127],
        'obj_types': ['Car'],
        'mode': mode
    }
    return TripletDetectionLoader(source, loader_params)
Beispiel #3
0
def get_loader():
    source = KITTISource('/home/ray/Data/KITTI/training',max_frames=10000)
    loader_params = {'crop_size':[255,255],'obj_types':['Car']}
    return AutoEncoderLoader(source,loader_params)
def get_loader():
    source = KITTISource('/home/ray/Data/KITTI/tracking/training',max_frames=10000)
    return SequenceVideoLoader(source,crop_size=[512,160])
Beispiel #5
0
def get_loader():
    source = KITTISource('/home/ray/Data/KITTI/training', max_frames=10000)
    return AutoEncoderLoader(source, crop_size=[255, 255], obj_types=['Car'])
Beispiel #6
0
def get_loader():
    source = KITTISource('/home/ray/Data/KITTI/training', max_frames=10000)
    return MultiObjectDetectionLoader(source,
                                      crop_size=[1024, 320],
                                      obj_types=classes,
                                      max_objects=100)