PP-ShiTuV2是一个实用的轻量级通用图像识别系统,主要由主体检测、特征学习和向量检索三个模块组成。
这里只利用特征学习和向量检索模块,特征提取时使用的都是主体突出的图像。
特征库建库操作
模型整体推理
cd deploy
python python/predict_system.py -c configs/inference_drink.yaml -o Global.infer_imgs="./drink_dataset_v2.0/test_images"
修改代码
去掉主体检测模型部分即可