设置首页收藏本站
开启左侧

怎么让同一种类别,用同种颜色框框住,求助

[复制链接]
卡可 发表于 2025-1-5 17:29:15 | 显示全部楼层 |阅读模式
  1. dev_update_off ()
  2. query_available_dl_devices (['runtime', 'runtime'], ['gpu', 'cpu'], DLDeviceHandles)
  3. if (|DLDeviceHandles| == 0)
  4.     throw ('No supported device found to continue this example.')
  5. endif
  6. DLDevice := DLDeviceHandles[0]
  7. PreprocessParamFileName := 'F:/model/model_训练-241224-113117_dl_preprocess_params.hdict'
  8. RetrainedModelFileName := 'F:/model/model_训练-241224-113117.hdl'
  9. BatchSizeInference := 1
  10. MinConfidence := 0.5
  11. MaxOverlap := 0.2
  12. MaxOverlapClassAgnostic := 0.7
  13. read_dl_model (RetrainedModelFileName, DLModelHandle)
  14. set_dl_model_param (DLModelHandle, 'batch_size', BatchSizeInference)
  15. set_dl_model_param (DLModelHandle, 'device', DLDevice)
  16. set_dl_model_param (DLModelHandle, 'min_confidence', MinConfidence)
  17. set_dl_model_param (DLModelHandle, 'max_overlap', MaxOverlap)
  18. set_dl_model_param (DLModelHandle, 'max_overlap_class_agnostic', MaxOverlapClassAgnostic)
  19. read_dict (PreprocessParamFileName, [], [], DLPreprocessParam)
  20. get_dl_model_param (DLModelHandle, 'class_names', ClassNames)
  21. get_dl_model_param (DLModelHandle, 'class_ids', ClassIDs)
  22. list_files ('F:/新项目1_images','files',ImageFiles)
  23. for Ind := 0 to |ImageFiles| - 1 by 1
  24.     read_image (Image, ImageFiles[Ind])
  25.     get_image_size(Image, Width, Height)
  26.     gen_dl_samples_from_images (Image, DLSampleBatch)
  27.     preprocess_dl_samples (DLSampleBatch, DLPreprocessParam)
  28.     apply_dl_model (DLModelHandle, DLSampleBatch, [], DLResultBatch)
  29.     dev_display(Image)   
  30.     for SampleIndex := 0 to BatchSizeInference - 1 by 1
  31.         DLResult := DLResultBatch[SampleIndex]   
  32.         get_dict_tuple (DLResult, 'bbox_row1', BboxRow1)
  33.         get_dict_tuple (DLResult, 'bbox_col1', BboxCol1)
  34.         get_dict_tuple (DLResult, 'bbox_row2', BboxRow2)
  35.         get_dict_tuple (DLResult, 'bbox_col2', BboxCol2)
  36.         get_dict_tuple (DLResult, 'bbox_confidence', BboxConfidences)
  37.         get_dict_tuple (DLResult,'bbox_class_name',classname)
  38.         zoomX := Width / 256.0
  39.         zoomY := Height / 256.0
  40.         BboxRow1 := zoomY * BboxRow1
  41.         BboxCol1 := zoomX * BboxCol1
  42.         BboxRow2 := zoomY * BboxRow2
  43.         BboxCol2 := zoomX * BboxCol2   
  44.         gen_rectangle2_contour_xld (BboxRectangle, 0.5 * (BboxRow1 + BboxRow2), 0.5 * (BboxCol1 + BboxCol2),0,0.5 * (BboxCol1 - BboxCol2),0.5 * (BboxRow1 - BboxRow2) )
  45.         dev_display(BboxRectangle)
  46.         dev_disp_text (classname,'image',BboxRow2, BboxCol1,'red', 'box_color', '#ffffff77')
  47.         dev_disp_text (BboxConfidences,'image',BboxRow1, BboxCol1,'black', 'box_color', '#ffffff77')
  48.         dev_disp_text ('Press Run (F5) to continue', 'window', 'bottom', 'right', 'black', [], [])
  49.         stop ()
  50.     endfor
  51. endfor
复制代码

屏幕截图 2025-01-05 151820.png


奖励计划已经开启,本站鼓励作者发布最擅长的技术内容和资源,流量变现就在现在,[点我]加入吧~~~Go
Fanxf 发表于 2025-1-6 16:18:46 | 显示全部楼层
dev_set_color
奖励计划已经开启,本站鼓励作者发布最擅长的技术内容和资源,流量变现就在现在,[点我]加入吧~~~Go
寤寐求之 发表于 2025-1-7 11:59:32 | 显示全部楼层
先给不同类的缺陷用不同的空区域装着gen_empty_obj ,后面显示的时候先dev_set_color 然后在dev_display区域
奖励计划已经开启,本站鼓励作者发布最擅长的技术内容和资源,流量变现就在现在,[点我]加入吧~~~Go
您需要登录后才可以回帖 登录 | 注册

本版积分规则

快速回复 返回顶部 返回列表