def init()

in src/retrieval_utils.py [0:0]

• 39 lines of code
• 1 McCabe index (conditional complexity)

def init(args, get_video_encoder_only=True, logger=None):

    # Loading Train data
    print("Loading training data")
    dataset = AVideoDataset(
        ds_name=args.dataset,
        root_dir=args.root_dir,
        mode='train',
        num_frames=args.clip_len,
        sample_rate=args.steps_bet_clips,
        num_train_clips=args.train_clips_per_video,
        train_crop_size=112,
        seed=None,
        fold=args.fold,
        colorjitter=False,
        temp_jitter=True,
        center_crop=False,
        target_fps=30,
        decode_audio=False,
    )

    print("Loading validation data")
    dataset_test = AVideoDataset(
        ds_name=args.dataset,
        root_dir=args.root_dir,
        mode='test',
        num_frames=args.clip_len,
        sample_rate=args.steps_bet_clips,
        num_spatial_crops=1,
        num_ensemble_views=args.train_clips_per_video,
        test_crop_size=112,
        seed=None,
        fold=args.fold,
        colorjitter=False,
        temp_jitter=True,
        center_crop=False,
        target_fps=30,
        decode_audio=False,
    )

    model = get_model(args, 
        get_video_encoder_only=get_video_encoder_only, logger=logger)
    return model, dataset, dataset_test