车道线检测laneatt 学习笔记

目录

图片检测可视化

---

图片检测可视化

import logging
import argparse
import os
import time

import cv2
import numpy as np
import torch

from lib.config import Config
from lib.runner import Runner
from lib.experiment import Experiment


def parse_args():
    parser = argparse.ArgumentParser(description="Train lane detector")
    parser.add_argument("--mode", choices=["test"], help="Train or test?")
    parser.add_argument("--exp_name", help="Experiment name", default='laneatt_r18_tusimple')
    parser.add_argument("--cfg",default='cfgs/laneatt_tusimp