近日,香港中文大學多媒體實驗室(MMLab)和南洋理工大學的研究者最近開源了一套統一的自監督學習代碼庫OpenSelfSup。OpenSelfSup使用PyTorch實現,支持基于分類、重建、聚類、memory bank、contrastive learning的多種自監督學習框架,目前收錄了NPID、MoCo、SimCLR等一系列表現較好的自監督表征學習方法,后續還將陸續跟進學術界最新算法。在這個框架下,每個算法被拆解為backbone、neck、head等多個可獨立設計的模塊,每個模塊均提供多個可選方案,開發者也可以自行設計各個模塊。