How to predict the class in the test dataset?

Line 273 in 5b1508d

class_type = test_dataset.type_list[load_batch_size * i + b]

Line 293 in 5b1508d

det_results[data_idx][class_type].append(output)

In the test_net_det.py , the class_type is from the validation test data label. But in the test dataset, we can't get this info. So How can we get the class in the test dataset.