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

Mionr bug: There is no `r_model_path` in `create_mtcnn_net()` #28

Open RicoSuaveGuapo opened 4 years ago

RicoSuaveGuapo commented 4 years ago

Just a small bug, there is no r_model_path in create_mtcnn_net. Therefore, get_Onet_train_data.py does not have to assign create_mtcnn_net with r_model_path anymore.

RicoSuaveGuapo commented 4 years ago

Also in line bboxes = create_mtcnn_net(image, 50, device, p_model_path=weight_path, o_model_path=None) should change to bboxes = create_mtcnn_net(image, [50,50], device, p_model_path=weight_path, o_model_path=None). Since min_lp_size in detect_pnet(pnet, image, min_lp_size, device) should either tuple or list with len 2.