License_Plate_Detection_Pytorch icon indicating copy to clipboard operation
License_Plate_Detection_Pytorch copied to clipboard

create_mtcnn_net does not give the same shapes for the bboxes.

Open danielchang1985 opened this issue 3 years ago • 0 comments

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)

danielchang1985 avatar Apr 10 '21 12:04 danielchang1985