提交 d36dfb09 编写于 作者: D dongshuilong

fix index_manager bug

上级 f64bec7f
Global:
infer_imgs: "./drink_dataset_v1.0/test_images/hongniu_1.jpg"
det_inference_model_dir: "/Users/dongshuilong/Desktop/Index_manager/picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer"
rec_inference_model_dir: "/Users/dongshuilong/Desktop/Index_manager/inference_taml"
rec_nms_thresold: 0.05
batch_size: 1
image_shape: [3, 640, 640]
threshold: 0.2
max_det_results: 5
label_list:
- foreground
use_gpu: True
enable_mkldnn: False
cpu_num_threads: 10
enable_benchmark: False
use_fp16: False
ir_optim: True
use_tensorrt: False
gpu_mem: 8000
enable_profile: False
DetPreProcess:
transform_ops:
- DetResize:
interp: 2
keep_ratio: false
target_size: [640, 640]
- DetNormalizeImage:
is_scale: true
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
- DetPermute: {}
DetPostProcess: {}
RecPreProcess:
transform_ops:
- ResizeImage:
size: 224
- NormalizeImage:
scale: 0.00392157
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
order: ''
- ToCHWImage:
RecPostProcess: null
# indexing engine config
IndexProcess:
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
image_root: "./drink_dataset_v1.0/gallery"
index_dir: "./drink_dataset_v1.0/index"
data_file: "./drink_dataset_v1.0/gallery/drink_label.txt"
index_operation: "new" # suported: "append", "remove", "new"
delimiter: "\t"
dist_type: "IP"
embedding_size: 512
batch_size: 32
return_k: 5
score_thres: 0.4
...@@ -131,7 +131,7 @@ class ClassifyUiContext(QtCore.QObject): ...@@ -131,7 +131,7 @@ class ClassifyUiContext(QtCore.QObject):
"分类下存在图片,请先移除图片") "分类下存在图片,请先移除图片")
else: else:
self.__imageListMgr.removeClassify(classify) self.__imageListMgr.removeClassify(classify)
self.setClassifyList(self.__imageListMgr.classifyList()) self.setClassifyList(self.__imageListMgr.classifyList)
def renemeClassify(self): def renemeClassify(self):
"""重命名分类""" """重命名分类"""
......
...@@ -9,7 +9,7 @@ class ImageEditClassifyDialog(QtWidgets.QDialog): ...@@ -9,7 +9,7 @@ class ImageEditClassifyDialog(QtWidgets.QDialog):
"""图像编辑分类对话框""" """图像编辑分类对话框"""
def __init__(self, parent, old_classify, classify_list): def __init__(self, parent, old_classify, classify_list):
super(ImageEditClassifyDialog, self).__init__(parent) super(ImageEditClassifyDialog, self).__init__(parent)
self.ui = mod.ui_imageeditclassifydialog.Ui_Dialog() self.ui = ui_imageeditclassifydialog.Ui_Dialog()
self.ui.setupUi(self) # 初始化主窗口界面 self.ui.setupUi(self) # 初始化主窗口界面
self.__oldClassify = old_classify self.__oldClassify = old_classify
self.__classifyList = classify_list self.__classifyList = classify_list
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册