/KW_ADAS

Advanced Dirver Assistance System

Primary LanguagePython

KW_ADAS

Advanced Dirver Assistance System

필요 library 설치

pip install -r requirements.txt

사용한 데이터 셋

-line segmentation

-detection

데이터 전처리

  • detection
python ./utils/detection_file_rename.py
python ./utils/detection_json_to_yolo.py
python ./utils/detection_dataset_split.py
  • segmentation
python ./utils/segmentation_map.py

yolo config file

python ./utils/detection_make_yolo_config.py

model train

  • detection
python yolo_train.py
  • segmentation
python ./pytorch-nested-unet/train.sh

model eval

  • detection
python yolo_eval.py
Model mAP50 Download
yolov8s 0.88 link
  • segmentation
python ./pytorch-nested-unet/val.py --name lane_segmentation
Model IoU Loss Download
U-Net++ 0.852 0.086 link

Result

python inference.py

Demo