Initial Commit (tested training, testing, and TRT conversion)

2024-10-20 17:01:07 +08:00
parent 86d2f311f8
commit 5738088bae
221 changed files with 59249 additions and 6 deletions
--- a/.gitignore
+++ b/.gitignore
@@ -0,0 +1,33 @@
+# Python
+dist
+*.egg-info
+*.so
+__pycache__
+
+# Pycharm
+.idea/
+
+# VScode
+.vscode/
+.vscode/*
+**/.vscode/*
+!.vscode/extensions.json
+
+# General
+_build/
+.sass-cache
+_posts
+_site
+
+# saved
+run/saved/*
+!run/saved/YOPO_1
+run/yopo_trt.pth
+externals/
+
+#logs
+run/utils/*.csv
+
+# dataset
+/run/yopo_sim/*
+