icdd-vsumm/config.py

import os
import torch


class ServerConfig(object):
    # 视频文件配置
    videos_root_path     : str = os.path.join('data', 'videos')
    videos_play_path     : str = os.path.join(videos_root_path, 'play')
    videos_upload_path   : str = os.path.join(videos_root_path, 'upload')
    videos_chunk_size    : int = 1024 * 1024 * 1024
    # 数据文件配置
    hdf5_root_path       : str = os.path.join('data', 'h5s')

    # TODO


class DSNetConfig(object):
    device                  : str   = 'cuda' if torch.cuda.is_available() else 'cpu'
    random_seed             : int   = 4906
    sample_rate             : int   = 15
    nms_thresh              : float = 0.5
    lambda_center           : float = 1.
    # TODO
    pass