cannot draw RBBOX during Inference

[shiv4ngi]

Hey, Thank you for the repo, I am using this repo to learn and train my own model on rotated images dataset. I trained the model on my images successfully and now trying to run some inference. I used test.py from tools folder with type=OBB and then parsed the pkl file using parse_result.py

I got txt files by class names with bounding boxes. I wanted to plot RBBOX on the image itself during inference and save plotted images, is there any way I can do this directly?

I tried --show parameter but it returns this error - assert bboxes.shape[1] == 4 or bboxes.shape[1] == 5

[dingjiansw101]

Please see the demo https://github.com/dingjiansw101/AerialDetection/blob/master/demo_large_image.py