add gkt

c5d6a51a · wanghailong · a57fa6f8 · c5d6a51a · c5d6a51a · c5d6a51a
Commit c5d6a51a authored Mar 25, 2024 by wanghailong
7 changed files
--- a/.gitignore
+++ b/.gitignore
@@ -23,6 +23,7 @@ data/
 ckpts
 data
 test/
+runs/
 val/
 work_dirs/
 develop-eggs/
@@ -150,6 +151,8 @@ cluster.sh
 #test
 *.onnx
 *.ckpts
+*.ckpt
+*.pkl
 *.hbir
 *.pt
 *.hbm

--- a/README.md
+++ b/README.md
--- a/projects/configs/maptrv2/maptrv2_nusc_r50_24ep_w_centerline_j5_gkt.py
+++ b/projects/configs/maptrv2/maptrv2_nusc_r50_24ep_w_centerline_j5_gkt.py
--- a/projects/mmdet3d_plugin/__init__.py
+++ b/projects/mmdet3d_plugin/__init__.py
@@ -11,3 +11,4 @@ from .models.opt.adamw import AdamW2
 from .bevformer import *
 from .maptr import *
 from .models.backbones.efficientnet import EfficientNet
+from .hat.gtk_transformer import GKTTransformer
\ No newline at end of file
--- a/projects/mmdet3d_plugin/hat/gtk_transformer.py
+++ b/projects/mmdet3d_plugin/hat/gtk_transformer.py
--- a/tools/maptr/to_onnx_j5_gkt.py
+++ b/tools/maptr/to_onnx_j5_gkt.py
--- a/tools/maptr/to_onnx_j5_gkt_cpu.py
+++ b/tools/maptr/to_onnx_j5_gkt_cpu.py