【文件属性】:
文件名称:OpenUnReID:PyTorch开源工具箱,用于无监督或领域自适应对象重新ID
文件大小:531KB
文件格式:ZIP
更新时间:2021-05-05 08:14:02
unsupervised-learning image-retrieval re-identification pseudo-labeling unsupervised-domain-adaptation
OpenUnReID
介绍
OpenUnReID是一个基于PyTorch的开源代码库,在对象重新ID任务的上下文中既可用于非监督学习( USL )也可用于非监督域自适应( UDA )。 它为基于伪标签的框架和基于域翻译的框架提供了强大的基线和多种最新方法,并提供了高度重构的代码。 它适用于Python> = 3.5和PyTorch> = 1.1 。
我们正在积极更新此存储库,不久将支持更多方法。 欢迎捐款。
主要特点
使用多个GPU和多台机器进行分布式培训和测试。
数据集,主干,损失等的各种组合具有高度的灵活性。
基于GPU的伪标签生成和k倒数重新排序速度非常快。
适用于任何骨干网的即插即用域特定的BatchNorm,还支持同步BN。
支持混合精度训练,从而实现更高的效率。
强大的集群基线,可为设计新方法提供高度可扩展性。
有关对象重新ID的USL和UDA问题的最新方法和性