Turoad/CLRNet

detect检测不出车道线

Closed this issue · 0 comments

使用官方模型 https://download.pytorch.org/models/resnet34-333f7ec4.pth 数据采用CULane
test_parameters = dict(conf_threshold=0.01, nms_thres=50, nms_topk=max_lanes) conf_threshold 已经改成0.01 其余保存不变
但是保存下来的图片没有画出车道线
在detect.py
def run(self, data):
data = self.preprocess(data)
data['lanes'] = self.inference(data)[0]
print("data['lanes']", data['lanes'], data['img_path'])
if self.cfg.show or self.cfg.savedir:
self.show(data)
return data
打印data['lanes']的值也为空,另外运行时data['lanes']会有变动,有时为空有时为2x2的np.array 但是保存图片还是没有车道线画出。