some questions about rotation_demo.py !

[FakerYFX]

@mjq11302010044
if the CLASSES = ('chinese', 'japenese',"English","germany"),I want to get the different class's result, how can I modify the code below:
//Visualize detections for each class
CONF_THRESH = conf
NMS_THRESH = 0.3
for cls_ind, cls in enumerate(CLASSES[1:]):
cls_ind += 1 # because we skipped background
cls_boxes = boxes[:, 5 * cls_ind:5 * (cls_ind + 1)] # D
cls_scores = scores[:, cls_ind]
dets = np.hstack((cls_boxes,
cls_scores[:, np.newaxis])).astype(np.float32)
keep = rotate_gpu_nms(dets, NMS_THRESH) # D
dets = dets[keep, :]
//dets = dets[0:20]
//dets[:, 4] = dets[:, 4] * 0.45
dets[:, 2] = dets[:, 2] / cfg.TEST.GT_MARGIN
dets[:, 3] = dets[:, 3] / cfg.TEST.GT_MARGIN
results = write_result_ICDAR(
im_file,
dets,
CONF_THRESH,
result_dir,
im_height,
im_width)
return results

[mjq11302010044]

Better to use a for loop to get predictions of different classes according to the bbox score