batch_indexs = [
idx - min_idx for idx in state['sample']['batch_anno_idxs']
]
sorted_segments = [state['output'][i] for i in batch_indexs]
state['sorted_segments_list'].extend(sorted_segments)
def on_test_end(state):
if config.VERBOSE:
state['progress_bar'].close()
print()
annotations = test_dataset.annotations
state['Rank@N,mIoU@M'], state['miou'] = eval_predictions(
state['sorted_segments_list'], annotations, verbose=True)
loss_message = '\ntest loss {:.4f}'.format(state['loss_meter'].avg)
print(loss_message)
state['loss_meter'].reset()
test_table = display_results(state['Rank@N,mIoU@M'], state['miou'],
'performance on testing set')
table_message = '\n' + test_table
print(table_message)
# save_scores(state['sorted_segments_list'], annotations, config.DATASET.NAME, args.split)
engine = Engine()
engine.hooks['on_test_start'] = on_test_start
engine.hooks['on_test_forward'] = on_test_forward
engine.hooks['on_test_end'] = on_test_end
engine.test(network, dataloader, args.split)
