车道线检测神经网络笔记

目录

2022-Ultra-Fast-Lane-Detection-v2

测试代码跑通 

2023 LVLane

3D车道线单目检测方法 ONCE-3DLanes

斑马线 检测：

边做检测，边做车道线检测的多任务网络。

2021年

LSTR: Lane Shape Prediction with Transformers

2.2 损失函数

方法介绍

LaneNet实时车道线检测和智能告警

---

2022-Ultra-Fast-Lane-Detection-v2

https://github.com/cfzd/Ultra-Fast-Lane-Detection-v2

测试代码跑通 

import torch, os, cv2
from utils.dist_utils import dist_print
import torch, os
from utils.common import merge_config, get_model
import tqdm
import torchvision.transforms as transforms
from data.dataset import LaneTestDataset

def pred2coords(pred, row_anchor, col_anchor, local_width = 1, original_image_width = 1640, original_image_height = 590):
    batch_size, num_grid_row, num_cls_row, num_lane_row = pred['loc_row'].shape
    batch_size, num_grid_col, num_cls_col, num_lane_col = pred['loc_col'].shape

    max_indices_