python hybridnets_test_videos.py --output demo_result
A6000
torch==2.0.1
torchvision==0.15.2
timm==0.6.7
python hybridnets_test_videos.py --output demo_result --cuda True
python hybridnets_test_videos.py --output demo_result --cuda True --float16 True
fps 6.216
python demo.py --source ../input/1.mp4 --device cpu --name video_1_output --nosave
fps 5.8
python demo.py --source ../input/1.mp4 --device 0 --name video_1_output --nosave
torch와 CUDA 버전 호환 문제 발생
정확히는 torch==1.7.0
과 CUDA==1.7
호환 안됨
INSTALLING PREVIOUS VERSIONS OF PYTORCH 호환 버전 확인
CUDA==9.2
,10.1
,10.2
,11.0
중 하나 필요
호환 버전 없음
CUDA 11.0
설치
환경변수 수정 후 버전 변경 확인
torch와 CUDA 버전 호환 문제 지속 발생
CUDA와 A6000 버전 호환 문제
torch 버전 변경
A6000
CUDA==11.7
torch==2.0.1
torchvision==0.15.2
fps 166
Traffic Object Detection
Model | Recall (%) | mAP@0.5 (%) |
---|---|---|
HybridNets | 92.8 | 77.3 |
YOLOPv2 | 91.1 | 83.4 |
Drivable Area Segmentation
Model | Recall (%) |
---|---|
HybridNets | 90.5 |
YOLOPv2 | 93.2 |
Lane Line Detection
Model | Accuracy (%) | Lane Line IoU (%) |
---|---|---|
HybridNets | 85.4 | 31.6 |
YOLOPv2 | 87.3 | 27.2 |
HybridNets: End-to-End Perception Network
YOLOPv2: Better, Faster, Stronger for Panoptic Driving Perception