62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
|
# --------------------------------------------------------
|
||
|
|
# mcan-vqa (Deep Modular Co-Attention Networks)
|
||
|
|
# Licensed under The MIT License [see LICENSE for details]
|
||
|
|
# Written by Yuhao Cui https://github.com/cuiyuhao1996
|
||
|
|
# --------------------------------------------------------
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
class PATH:
|
||
|
|
def __init__(self, EXP_NAME, DATASET_PATH):
|
||
|
|
# name of the experiment
|
||
|
|
self.EXP_NAME = EXP_NAME
|
||
|
|
|
||
|
|
# Dataset root path
|
||
|
|
self.DATASET_PATH = DATASET_PATH
|
||
|
|
|
||
|
|
# Bottom up features root path
|
||
|
|
self.FRAMES = os.path.join(DATASET_PATH, 'frame_feat/')
|
||
|
|
self.CLIPS = os.path.join(DATASET_PATH, 'clip_feat/')
|
||
|
|
|
||
|
|
|
||
|
|
def init_path(self):
|
||
|
|
self.QA_PATH = {
|
||
|
|
'train': self.DATASET_PATH + 'train_qa.json',
|
||
|
|
'val': self.DATASET_PATH + 'val_qa.json',
|
||
|
|
'test': self.DATASET_PATH + 'test_qa.json',
|
||
|
|
}
|
||
|
|
self.C3D_PATH = self.DATASET_PATH + 'c3d.pickle'
|
||
|
|
|
||
|
|
if self.EXP_NAME not in os.listdir('./'):
|
||
|
|
os.mkdir('./' + self.EXP_NAME)
|
||
|
|
os.mkdir('./' + self.EXP_NAME + '/results')
|
||
|
|
self.RESULT_PATH = './' + self.EXP_NAME + '/results/result_test/'
|
||
|
|
self.PRED_PATH = './' + self.EXP_NAME + '/results/pred/'
|
||
|
|
self.CACHE_PATH = './' + self.EXP_NAME + '/results/cache/'
|
||
|
|
self.LOG_PATH = './' + self.EXP_NAME + '/results/log/'
|
||
|
|
self.TB_PATH = './' + self.EXP_NAME + '/results/tensorboard/'
|
||
|
|
self.CKPTS_PATH = './' + self.EXP_NAME + '/ckpts/'
|
||
|
|
|
||
|
|
if 'result_test' not in os.listdir('./' + self.EXP_NAME + '/results'):
|
||
|
|
os.mkdir('./' + self.EXP_NAME + '/results/result_test/')
|
||
|
|
|
||
|
|
if 'pred' not in os.listdir('./' + self.EXP_NAME + '/results'):
|
||
|
|
os.mkdir('./' + self.EXP_NAME + '/results/pred/')
|
||
|
|
|
||
|
|
if 'cache' not in os.listdir('./' + self.EXP_NAME + '/results'):
|
||
|
|
os.mkdir('./' + self.EXP_NAME + '/results/cache')
|
||
|
|
|
||
|
|
if 'log' not in os.listdir('./' + self.EXP_NAME + '/results'):
|
||
|
|
os.mkdir('./' + self.EXP_NAME + '/results/log')
|
||
|
|
|
||
|
|
if 'tensorboard' not in os.listdir('./' + self.EXP_NAME + '/results'):
|
||
|
|
os.mkdir('./' + self.EXP_NAME + '/results/tensorboard')
|
||
|
|
|
||
|
|
if 'ckpts' not in os.listdir('./' + self.EXP_NAME):
|
||
|
|
os.mkdir('./' + self.EXP_NAME + '/ckpts')
|
||
|
|
|
||
|
|
|
||
|
|
def check_path(self):
|
||
|
|
raise NotImplementedError
|
||
|
|
|