main

Author: StevenGrove

README.md

@@ -1,6 +1,6 @@
 # TreeFilter-Torch
 This project provides a cuda implementation for "[Learnable Tree Filter for Structure-preserving
-Feature Transform](https://megvii-my.sharepoint.cn/:b:/g/personal/songlin_megvii_com/EfbrITIdvqBCu-SaW9gZOHQBFIkcIisB6-FyO9SzzrZyPQ?e=YI06YP)" on PyTorch. Multiple semantic segmentation experiments are reproduced to verify the effectiveness of tree filtering module on PASCAL VOC2012 and Cityscapes. For the reason that the experiments in the paper were conducted using internal framework, this project reimplements them on PyTorch and reports detailed comparisons below. In addition, many thanks to [TorchSeg](https://github.com/ycszen/TorchSeg).
+Feature Transform](https://arxiv.org/pdf/1909.12513.pdf)" on PyTorch. Multiple semantic segmentation experiments are reproduced to verify the effectiveness of tree filtering module on PASCAL VOC2012 and Cityscapes. For the reason that the experiments in the paper were conducted using internal framework, this project reimplements them on PyTorch and reports detailed comparisons below. In addition, many thanks to [TorchSeg](https://github.com/ycszen/TorchSeg).
 
 ![introduce image](demo/introduce.png)
 
@@ -74,16 +74,23 @@ For detailed usage, please refer to the [Training](https://github.com/ycszen/Tor
 Please cite the learnable tree filter in your publications if it helps your research. 
 
 ```
-The pre-printed version has been submitted to Arxiv and is awaiting public.
+@misc{song2019learnable,
+    title={Learnable Tree Filter for Structure-preserving Feature Transform},
+    author={Lin Song and Yanwei Li and Zeming Li and Gang Yu and Hongbin Sun and Jian Sun and Nanning Zheng},
+    year={2019},
+    eprint={1909.12513},
+    archivePrefix={arXiv},
+    primaryClass={cs.CV}
+}
 ```
 
 Please cite this project in your publications if it helps your research. 
 ```
 @misc{treefilter-torch,
-  author =       {Song, Lin},
-  title =        {TreeFiler-Torch},
-  howpublished = {\url{https://github.com/StevenGrove/TreeFilter-Torch}},
-  year =         {2019}
+    author = {Song, Lin},
+    title = {TreeFiler-Torch},
+    howpublished = {\url{https://github.com/StevenGrove/TreeFilter-Torch}},
+    year ={2019}
 }
 ```