image_to_pixle_params_yoloSAM/main/readme.md

1.2 KiB
Raw Blame History

下载环境

pip install -r requirements.txt

数据集位置

demo/image 中存在3张front.jpg , rear.jpgside.jpg 分别代表汽车前方、后方、侧方

运行代码

  1. python3 u2net_saliency.py 将原图经过模型生成显著图
  2. python3 main.py 将显著图进行轮胎圆轴距检测,长宽高检测
  3. python3 point.py 将进行接触点检测

实际效果

原始图片

front rear side

最后结果

外接矩形和轴距检测

detected_centers_original front_size_lines_orig side_size_lines_orig rear_size_lines_orig

周围接触点检测

side_output_rgb

front_output_rgb

rear_output_rgb