import paddleclas

model = paddleclas.PaddleClas(build_gallery=True,model_name='PP-ShiTuV2',index_dir='./drink_dataset_v2.0/index',gallery_image_root='./drink_dataset_v2.0/gallery/',gallery_data_file='./drink_dataset_v2.0/gallery/drink_label.txt')

# paddleclas --build_gallery=True --model_name="PP-ShiTuV2" -o IndexProcess.image_root=./drink_dataset_v2.0/gallery/ -o IndexProcess.index_dir=./drink_dataset_v2.0/index -o IndexProcess.data_file=./drink_dataset_v2.0/gallery/drink_label.txt