hirotomusiker / CLRerNet

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

Feature/laneiou loss #17

Closed hirotomusiker closed 1 year ago

hirotomusiker commented 1 year ago

Unit test results:

Name                                         Stmts   Miss  Cover
----------------------------------------------------------------
libs/core/anchor/__init__.py                     1      0   100%
libs/core/anchor/anchor_generator.py            23      0   100%
libs/core/bbox/match_costs/__init__.py           1      0   100%
libs/core/bbox/match_costs/match_cost.py        97     21    78%
libs/datasets/__init__.py                        2      0   100%
libs/datasets/culane_dataset.py                 74     37    50%
libs/datasets/metrics/culane_metric.py          89     73    18%
libs/datasets/pipelines/__init__.py              3      0   100%
libs/datasets/pipelines/alaug.py               154    138    10%
libs/datasets/pipelines/compose.py              31     19    39%
libs/datasets/pipelines/lane_formatting.py      19     12    37%
libs/models/__init__.py                          5      0   100%
libs/models/backbones/__init__.py                1      0   100%
libs/models/backbones/dla.py                   220     79    64%
libs/models/dense_heads/__init__.py              1      0   100%
libs/models/dense_heads/clrernet_head.py       145      7    95%
libs/models/detectors/__init__.py                1      0   100%
libs/models/detectors/clrernet.py               15      1    93%
libs/models/layers/__init__.py                   1      0   100%
libs/models/layers/attentions.py                73      0   100%
libs/models/losses/__init__.py                   1      0   100%
libs/models/losses/iou_loss.py                  48      0   100%
libs/models/necks/__init__.py                    1      0   100%
libs/models/necks/clrernet_fpn.py               36      1    97%
libs/utils/lane_utils.py                        33     24    27%
libs/utils/visualizer.py                        28     20    29%
----------------------------------------------------------------
TOTAL                                         1103    432    61%

========== 6 passed, 13 warnings in 3.60s ==========