voldemortX / pytorch-auto-drive

PytorchAutoDrive: Segmentation models (ERFNet, ENet, DeepLab, FCN...) and Lane detection models (SCNN, RESA, LSTR, LaneATT, BézierLaneNet...) based on PyTorch with fast training, visualization, benchmarking & deployment help
BSD 3-Clause "New" or "Revised" License
824 stars 137 forks source link

Run lane_video. py with resnet18_bezierlanene_tusimple.aug1b_20211109.pt , save the video without detection results? #173

Open 1623021453 opened 4 months ago

1623021453 commented 4 months ago

Hello, when I run lane_video. py with python tools/vis/lane_video.py --video-path=/media/user/新加卷/zxc/code/Vehicle-CV-ADAS/TrafficLaneDetector/temp/example.mp4 --save-path=/media/user/新加卷/zxc_ubuntu/code/pytorch-auto-drive/output/test.mp4 --config=/media/user/ 新加卷/zxc_ubuntu/code/pytorch-auto-drive/configs/lane_detection/bezierlanenet/resnet18_tusimple_aug1b.py --style=bezier --checkpoint=/media/user/新加卷/zxc_ubuntu/code/pytorch-auto-drive/resnet18_bezie rlanenet_tusimple_aug1b_20211109.pt , save the video without detection results.Who can tell me why?

voldemortX commented 4 months ago

@1623021453 maybe add some debug log to verify if the model indeed has outputs

1623021453 commented 4 months ago

谢谢回复,确实没有输出,keypoints是空的,我用resnet18_bezie rlanenet_tusimple_aug1b_20211109.pt 测tusimple的图片数据是有检测结果的,测视频没有检测结果是什么原因呢?我用erfnet_baseline_tusimple_20210424.pt测同一个视频,检测结果很好

voldemortX commented 4 months ago

@1623021453 这个可能得具体看下是解析后处理有bug,还是模型原始结果没有。你看的只是keypoints是空的对吧。模型原始的score和控制点有么