# model load configuse_gpu 0gpu_id 0gpu_mem 4000cpu_math_library_num_threads 30use_mkldnn 0# det configmax_side_len 600det_db_thresh 07det_db_box_thresh 03det_db_unclip_ratio 16use_polygon_scor
模型加载配置
- use_gpu:是否使用GPU进行训练,0表示不使用GPU,1表示使用GPU。
- gpu_id:指定使用的GPU的ID。
- gpu_mem:指定GPU的使用内存大小。
- cpu_math_library_num_threads:指定CPU的数学库线程数。
- use_mkldnn:是否使用MKL-DNN库。
目标检测配置
- max_side_len:图像的最大边长。
- det_db_thresh:文本检测的二值化阈值。
- det_db_box_thresh:检测到的文本框的阈值。
- det_db_unclip_ratio:文本框的膨胀比例。
- use_polygon_score:是否使用多边形得分。
- det_model_dir:目标检测模型的存储路径。
分类配置
- use_angle_cls:是否使用文本角度分类。
- cls_model_dir:分类模型的存储路径。
- cls_thresh:分类结果的阈值。
文本识别配置
- rec_model_dir:文本识别模型的存储路径。
- char_list_file:字符列表文件的路径。
显示检测结果
- visualize:是否显示检测结果。
使用TensorRT
- use_tensorrt:是否使用TensorRT进行推理。
- use_fp16:是否使用FP16数据类型。
原文地址: https://www.cveoy.top/t/topic/i2Az 著作权归作者所有。请勿转载和采集!