From 1b4d72c04cddfa85fe2c4348a49313c90469e645 Mon Sep 17 00:00:00 2001 From: Yigit Baran Can <37714607+ybarancan@users.noreply.github.com> Date: Wed, 1 Dec 2021 14:39:12 +0100 Subject: [PATCH] Update README.md --- README.md | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/README.md b/README.md index e565ced..6bc7b61 100644 --- a/README.md +++ b/README.md @@ -25,6 +25,13 @@ Nuscenes trained Polygon-RNN based model is at: https://data.vision.ee.ethz.ch/ Nuscenes trained Transformer based model is at: https://data.vision.ee.ethz.ch/cany/STSU/transformer.pth +## Metrics + +The implementation of the metrics can be found in src/utils/confusion.py. +Please refer to the paper for explanations on the metrics. + +![Metrics figure](metric.png) + ### Additional Links - Polygon-RNN: https://github.com/fidler-lab/polyrnn-pp