search

dingjiansw101 / AerialDetection

Apache License 2.0
798 stars 184 forks source link

cannot draw RBBOX during Inference #5

Open shiv4ngi opened 4 years ago

shiv4ngi commented 4 years ago

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 commented 4 years ago

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