36 lines
934 B
Python
36 lines
934 B
Python
#!/usr/bin/env python3
|
||
|
||
import cv2
|
||
import sys
|
||
import os
|
||
|
||
# 添加父目录到路径，以便能够导入utils
|
||
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
||
from utils.decode_arrow import detect_arrow_direction, visualize_arrow_detection
|
||
|
||
def main():
|
||
# 检查命令行参数
|
||
if len(sys.argv) < 2:
|
||
print("使用方法: python test_arrow.py <图像路径>")
|
||
sys.exit(1)
|
||
|
||
# 获取图像路径
|
||
image_path = sys.argv[1]
|
||
|
||
# 检查文件是否存在
|
||
if not os.path.exists(image_path):
|
||
print(f"错误: 文件 '{image_path}' 不存在")
|
||
sys.exit(1)
|
||
|
||
print(f"正在处理图像: {image_path}")
|
||
|
||
# 检测箭头方向
|
||
direction = detect_arrow_direction(image_path)
|
||
print(f"检测到的箭头方向: {direction}")
|
||
|
||
# 可视化检测过程
|
||
visualize_arrow_detection(image_path)
|
||
|
||
if __name__ == "__main__":
|
||
main() |