[发明专利]一种宠物识别方法

权利要求书

1.一种宠物识别方法，其特征在于，所述方法包括以下步骤：

1）使用yolov5作为宠物识别模型；

2）采集若干含有宠物图像的图片；

3）使用语义分割模型将图片中的宠物和背景分割开来，按照分割的结果把含有宠物的图片另存为具有透明通道的png图片，将背景的部分变为透明，使图片只保留宠物部分；

4）对图片进行数据增强处理，具体为：读取原图片中宠物打标的框的尺寸，按照所述尺寸以及项目要求设定增强后的图片的尺寸，并使用对应场景图作为背景，再将宠物原图贴在背景上；

5）使用程序按照一定随机的系数微调数据增强后的图片，具体为调整图片的色度、亮度和对比度；

6）将调整后的图片输入宠物识别模型并使用Generalized Focal Loss损失函数来训练宠物识别模型，得到训练好的宠物识别模型；

7）采集含有宠物图像的图片并输入训练好的宠物识别模型中；

8）获取采集的图片中宠物图像的特征，计算与训练好的宠物识别模型内存储的宠物的特征的相似度，输出宠物识别结果。

2.根据权利要求1所述的方法，其特征在于，步骤5）中使用python的pillow实现对图片色度、亮度和对比度的调整。