search

hirotomusiker / CLRerNet

The official implementation of "CLRerNet: Improving Confidence of Lane Detection with LaneIoU"
Apache License 2.0
181 stars 19 forks source link

how to set test batchsize? #51

Open RDXiaoLu opened 1 week ago

RDXiaoLu commented 1 week ago

I need test batchsize set 16. but how to do this?

hirotomusiker commented 1 week ago

I can't check it right now but could you check if it works by changing the following part? https://github.com/hirotomusiker/CLRerNet/blob/2b5632b55871b321080cce8069b01297b8134739/tools/test.py#L165