Merge branch 'feature/DeFlow'

Author: Kin-Zhang

1_train.py

@@ -3,8 +3,9 @@
 # Copyright (C) 2023-now, RPL, KTH Royal Institute of Technology
 # Author: Qingwen Zhang  (https://kin-zhang.github.io/)
 #
-# This work is licensed under the terms of the MIT license.
-# For a copy, see <https://opensource.org/licenses/MIT>.
+# This file is part of DeFlow (https://github.com/KTH-RPL/DeFlow).
+# If you find this repo helpful, please cite the respective publication as 
+# listed on the above website.
 
 # Description: Train Model
 """

2_eval.py

@@ -3,8 +3,9 @@
 # Copyright (C) 2023-now, RPL, KTH Royal Institute of Technology
 # Author: Qingwen Zhang  (https://kin-zhang.github.io/)
 #
-# This work is licensed under the terms of the MIT license.
-# For a copy, see <https://opensource.org/licenses/MIT>.
+# This file is part of DeFlow (https://github.com/KTH-RPL/DeFlow).
+# If you find this repo helpful, please cite the respective publication as 
+# listed on the above website.
 
 # Description: Output the evaluation results, go for local evaluation or online evaluation
 """

3_vis.py

@@ -3,8 +3,9 @@
 # Copyright (C) 2023-now, RPL, KTH Royal Institute of Technology
 # Author: Qingwen Zhang  (https://kin-zhang.github.io/)
 #
-# This work is licensed under the terms of the MIT license.
-# For a copy, see <https://opensource.org/licenses/MIT>.
+# This file is part of DeFlow (https://github.com/KTH-RPL/DeFlow).
+# If you find this repo helpful, please cite the respective publication as 
+# listed on the above website.
 
 # Description: produce flow based on model predict and write into the dataset, 
 #              then use tests/scene_flow.py flow_mode="flow_est" to visualize the flow.

dataprocess/extract_av2.py

@@ -3,8 +3,9 @@
 # Copyright (C) 2023-now, RPL, KTH Royal Institute of Technology
 # Author: Qingwen Zhang  (https://kin-zhang.github.io/)
 #
-# This work is licensed under the terms of the MIT license.
-# For a copy, see <https://opensource.org/licenses/MIT>.
+# This file is part of DeFlow (https://github.com/KTH-RPL/DeFlow).
+# If you find this repo helpful, please cite the respective publication as 
+# listed on the above website.
 
 # Description: Preprocess Data, save as h5df format for faster loading
 # Reference: 

dataprocess/extract_waymo.py

@@ -2,7 +2,11 @@
 # Created: 2024-02-26 12:42
 # Copyright (C) 2024-now, RPL, KTH Royal Institute of Technology
 # Author: Qingwen Zhang  (https://kin-zhang.github.io/), Tianshuai Hu (thuaj@connect.ust.hk)
-
+# 
+# This file is part of SeFlow (https://github.com/KTH-RPL/SeFlow).
+# If you find this repo helpful, please cite the respective publication as 
+# listed on the above website.
+# 
 # Description: Preprocess Data, save as h5df format for faster loading
 # This one is for Waymo dataset, refer a lot to 
 # Kyle's ZeroFlow code: https://github.com/kylevedder/zeroflow/tree/master/data_prep_scripts/waymo

scripts/network/loss_func.py

@@ -2,7 +2,11 @@
 # Created: 2023-07-17 00:00
 # Copyright (C) 2023-now, RPL, KTH Royal Institute of Technology
 # Author: Qingwen Zhang  (https://kin-zhang.github.io/)
-#
+# 
+# This file is part of DeFlow (https://github.com/KTH-RPL/DeFlow).
+# If you find this repo helpful, please cite the respective publication as 
+# listed on the above website.
+# 
 # Description: Define the loss function for training.
 """
 import torch

scripts/network/models/deflow.py

@@ -4,8 +4,9 @@
 # Copyright (C) 2023-now, RPL, KTH Royal Institute of Technology
 # Author: Qingwen Zhang  (https://kin-zhang.github.io/)
 #
-# This work is licensed under the terms of the MIT license.
-# For a copy, see <https://opensource.org/licenses/MIT>.
+# This file is part of DeFlow (https://github.com/KTH-RPL/DeFlow).
+# If you find this repo helpful, please cite the respective publication as 
+# listed on the above website.
 """
 
 import torch.nn as nn

scripts/pl_model.py

@@ -4,9 +4,10 @@
 # Copyright (C) 2023-now, RPL, KTH Royal Institute of Technology
 # Author: Qingwen Zhang  (https://kin-zhang.github.io/)
 #
-# This work is licensed under the terms of the MIT license.
-# For a copy, see <https://opensource.org/licenses/MIT>.
-
+# This file is part of DeFlow (https://github.com/KTH-RPL/DeFlow).
+# If you find this repo helpful, please cite the respective publication as 
+# listed on the above website.
+# 
 # Description: Model Wrapper for Pytorch Lightning
 
 """

tests/scene_flow.py

@@ -3,9 +3,10 @@
 # Copyright (C) 2023-now, RPL, KTH Royal Institute of Technology
 # Author: Qingwen Zhang  (https://kin-zhang.github.io/)
 #
-# This work is licensed under the terms of the MIT license.
-# For a copy, see <https://opensource.org/licenses/MIT>.
-
+# This file is part of DeFlow (https://github.com/KTH-RPL/DeFlow).
+# If you find this repo helpful, please cite the respective publication as 
+# listed on the above website.
+# 
 # Description: view scene flow dataset after preprocess.
 """