目录

图片检测可视化

---

图片检测可视化

import logging
import argparse
import os
import timeimport cv2
import numpy as np
import torchfrom lib.config import Config
from lib.runner import Runner
from lib.experiment import Experimentdef 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