模型加载配置

  • 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数据类型。
# 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

原文地址: https://www.cveoy.top/t/topic/i2Az 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录