caption, qids, questions, answers = parseInputFile(params['inputFile']) idx = np.array(qids, dtype='int') #inputTestSel = data['testData'][0][idx] #targetTestSel = data['testData'][1][idx] imgids = qids #imgids = inputTestSel[:, 0, 0] inputTest = prep.combine(\ prep.lookupQID(questions, data['questionDict'], maxlen), imgids) targetTest = prep.lookupAnsID(answers, data['ansDict']) questionTypeArray = data['questionTypeArray'][idx] print('Running models...') modelOutputs = ie.runAllModels( inputTest, questionTypeArray, params['models'], params['resultsFolder'], params['dataset'], params['dataFolder']) # Render if not os.path.exists(params['outputFolder']): os.makedirs(params['outputFolder']) if params['format'] == 'html': print('Rendering HTML...') pages = ir.renderHtml( inputTest, targetTest, data['questionIdict'], data['ansIdict'], urlDict,
-k: Top K answers -dataset: DAQUAR/COCO-QA dataset """ params = parseComparativeParams(sys.argv) print('Loading test data...') urlDict = loadImgUrl(params['dataset'], params['dataFolder']) data = it.loadDataset(params['dataFolder']) if len(params['models']) > 0: print('Running models...') singleModel = [params['models'][0]] modelOutputs = ie.runAllModels( data['testData'][0], data['questionTypeArray'], singleModel, params['resultsFolder'], params['dataset'], params['dataFolder']) else: modelOutputs = None # Render print('Rendering HTML to %s' % params['outputFolder']) if not os.path.exists(params['outputFolder']): os.makedirs(params['outputFolder']) pages = renderHtml( data['testData'][0], data['testData'][1], data['questionIdict'],