Class Prior-Free Positive-Unlabeled Learning with Taylor Variational Loss for Hyperspectral Remote Sensing Imagery
by Hengwei Zhao, Xinyu Wang, and Yanfei Zhong
This is an official implementation of T-HOneCls in our ICCV 2023 paper.
- Class prior-free PU learning for limited labeled hyperspectral imagery
- T-HOneCls achieves state-of-the-art results on 7 datasets (21 tasks in total)
- pytorch >= 1.13.1
- GDAL ==3.4.1
1.Modify the data path in the configuration file (./configs/X/XX/XXX.py).
The hyperspectral data can be obtained from the Link
(password:sqyy)
2.Training and testing
sh scripts/HongHu.sh
sh scripts/LongKou.sh
sh scripts/HanChuan.sh
If you use T-HOneCls in your research, please cite the following paper:
@InProceedings{Zhao_2023_ICCV,
author = {Zhao, Hengwei and Wang, Xinyu and Li, Jingtao and Zhong, Yanfei},
title = {Class Prior-Free Positive-Unlabeled Learning with Taylor Variational Loss for Hyperspectral Remote Sensing Imagery},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
month = {October},
year = {2023},
pages = {16827-16836}}
@article{ZHAO2022328,
title = {Mapping the distribution of invasive tree species using deep one-class classification in the tropical montane landscape of Kenya},
journal = {ISPRS Journal of Photogrammetry and Remote Sensing},
volume = {187},
pages = {328-344},
year = {2022},
issn = {0924-2716},
doi = {https://doi.org/10.1016/j.isprsjprs.2022.03.005},
url = {https://www.sciencedirect.com/science/article/pii/S0924271622000715},
author = {Hengwei Zhao and Yanfei Zhong and Xinyu Wang and Xin Hu and Chang Luo and Mark Boitt and Rami Piiroinen and Liangpei Zhang and Janne Heiskanen and Petri Pellikka}}
@ARTICLE{10174705,
author={Zhao, Hengwei and Zhong, Yanfei and Wang, Xinyu and Shu, Hong},
journal={IEEE Transactions on Geoscience and Remote Sensing},
title={One-Class Risk Estimation for One-Class Hyperspectral Image Classification},
year={2023},
volume={},
number={},
pages={1-1},
doi={10.1109/TGRS.2023.3292929}}
T-HOneCls can be used for academic purposes only, and any commercial use is prohibited.