xuexingyu24 / License_Plate_Detection_Pytorch

A two stage lightweight and high performance license plate recognition in MTCNN and LPRNet
Other
641 stars 171 forks source link

Variable box size for STN? #27

Open gryphus1153 opened 4 years ago

gryphus1153 commented 4 years ago

Is it possible to use different sized box for STN? It gives an exception when changing the "im = cv2.resize(img_box, (94, 24), interpolation=cv2.INTER_CUBIC)" dimensions from (94,24) to any other size i.e. (94,25).

Exception given: Exception has occurred: RuntimeError Sizes of tensors must match except in dimension 2. Got 5 and 4