add comment

Author: DylanWusee

evaluate.py

@@ -1,3 +1,9 @@
+"""
+Evaluation
+Author: Wenxuan Wu
+Date: May 2020
+"""
+
 import argparse
 import sys 
 import os 
@@ -86,7 +92,7 @@ def main():
     model.load_state_dict(torch.load(pretrain))
     print('load model %s'%pretrain)
     logger.info('load model %s'%pretrain)
-    
+
     model.cuda()
 
     epe3ds = AverageMeter()

evaluation_utils.py

@@ -1,3 +1,17 @@
+"""
+Evaluation metrics
+Borrowed from HPLFlowNet
+Date: May 2020
+
+@inproceedings{HPLFlowNet,
+  title={HPLFlowNet: Hierarchical Permutohedral Lattice FlowNet for
+Scene Flow Estimation on Large-scale Point Clouds},
+  author={Gu, Xiuye and Wang, Yijie and Wu, Chongruo and Lee, Yong Jae and Wang, Panqu},
+  booktitle={Computer Vision and Pattern Recognition (CVPR), 2019 IEEE International Conference on},
+  year={2019}
+}
+"""
+
 import numpy as np
 
 

models.py

@@ -1,3 +1,9 @@
+"""
+PointPWC-Net model and losses
+Author: Wenxuan Wu
+Date: May 2020
+"""
+
 import torch.nn as nn
 import torch
 import numpy as np

pointconv_util.py

@@ -1,3 +1,9 @@
+"""
+PointConv util functions
+Author: Wenxuan Wu
+Date: May 2020
+"""
+
 import torch 
 import torch.nn as nn