车道线检测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