HybridNets

CPU

python hybridnets_test_videos.py --output demo_result

A6000
torch==2.0.1 torchvision==0.15.2 timm==0.6.7

GPU

python hybridnets_test_videos.py --output demo_result --cuda True

python hybridnets_test_videos.py --output demo_result --cuda True --float16 True

result

fps 6.216

YOLOPv2

CPU

python demo.py --source ../input/1.mp4 --device cpu --name video_1_output --nosave

result

fps 5.8

GPU

python demo.py --source ../input/1.mp4 --device 0 --name video_1_output --nosave

dependency

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 버전 변경

result

A6000
CUDA==11.7 torch==2.0.1 torchvision==0.15.2

fps 166

HybridNets vs YOLOPv2

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

ref

HybridNets: End-to-End Perception Network
YOLOPv2: Better, Faster, Stronger for Panoptic Driving Perception