给定一张图片,检测出图中车牌的位置并输出车的类型(比如小汽车,挂车,新能源车等)。
本模型是以自底向上的方式: 1)首先识别出车牌的中心点;2)基于中心点回归出车牌的bbox;3)基于车牌中心点识别出车类型。模型介绍,详见:
输入图片,模型自动检测出所有车牌并给出对应的车型。用户可以自行尝试各种输入图片。具体调用方式请参考代码示例。
在安装完成ModelScope之后即可使用license-plate-detection的能力。
测试时的主要预处理和后处理如下:
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
license_plate_detection = pipeline(Tasks.license_plate_detection, model='damo/cv_resnet18_license-plate-detection_damo')
result = license_plate_detection('https://modelscope.oss-cn-beijing.aliyuncs.com/test/images/license_plate_detection.jpg')
print(result)
本模型利用imagenet预训练参数进行初始化,然后在训练数据集上进行训练。