search

mjq11302010044 / RRPN_pytorch

RRPN in pytorch 1.0 ----- Arbitrary-Oriented Scene Text Detection via Rotation Proposals
MIT License
262 stars 57 forks source link

A problem in transfrom #59

Open wnynotfaker opened 4 years ago

wnynotfaker commented 4 years ago

hello,I had a bug in class RandomRotation(object). In def rotate_boxes(self, target, angle), a error occured in "pts_ctr = np.squeeze(pts_ctr, axis=-1) + np.tile((im_width / 2, im_height / 2), (gt_boxes.shape[0], 1))". The valueerror is that cannot select an axis to squeeze out which has size not equal to one.

mjq11302010044 commented 4 years ago

@wnynotfaker Hi, you may have a wrong format of the input rotated box, please check if you have the correct format. Or you can try comment the rotate_boxes in the transform list to see if the code can run.

wnynotfaker commented 4 years ago

@wnynotfaker Hi, you may have a wrong format of the input rotated box, please check if you have the correct format. Or you can try comment the rotate_boxes in the transform list to see if the code can run.

Thank you for your reply. Finally I found out that I had some errors when modifying the data-augument code