search

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

create_mtcnn_net does not give the same shapes for the bboxes. #37

Open danielchang1985 opened 3 years ago

danielchang1985 commented 3 years ago

Hi I am running this block of codes but the shapes I am getting for bboxes1 and bboxes2 are different. Is there a reason to this and how can I correct it?

image1 = cv2.imread('test/6.jpg')
image2 = cv2.imread('train_data/train/class1/CB6692S.jpg')

input1 = np.copy(image1)
input2 = np.copy(image2)

bboxes1 = create_mtcnn_net(input1, (50, 15), device, p_model_path='MTCNN/weights/pnet_Weights', 
          o_model_path='MTCNN/weights/onet_Weights')
bboxes2 = create_mtcnn_net(input2, (50, 15), device, p_model_path='MTCNN/weights/pnet_Weights', 
          o_model_path='MTCNN/weights/onet_Weights')

bboxes1.shape (1,5) bboxes2.shape (0,5)