生成车道线分割数据集,图片和标注

import os
import random

import cv2
import numpy as np
import json

def distance(point1, point2):
    return np.sqrt((point1[0] - point2[0]) ** 2 + (point1[1] - point2[1]) ** 2)

def line_intersections(point1, point2, img_w, img_h):
    x1, y1 = point1
    x2, y2 = point2
    intersections = []

    if x1 != x2:
        # 左边界 x=0
        t = (0 - x1) / (x2 - x1)
        y = y1 + t * (y2 - y1)
        if 0 <= y <= img_h:
            intersections.append((0, int(y)))

        # 右边界 x=img_w
        t = (img_w - x1) / (x2 - x1)
        y = y1 + t * (y2 - y1)
        if 0 <= y <= img_h:
            intersections.append((img_w, int(y)))

    if y1 != y2:
        # 上边界 y=0
        t = (0 - y1) / (y2 - y1)
        x = x1 + t * (x2 - x1)
        if 0 <= x <= img_w:
            intersections.append((int(x), 0))

        # 下边界 y=img_h
  

你可能感兴趣的:(python宝典,1024程序员节)