Self-Supervised Learningのlibrary比較
- まだサポートされているか
- pipから簡単にinstall出来るか
- maintainerは信頼出来るか
- training pipelineのlibraryは何か
- comercial use出来るか
last-update | pip package | maintainer | training-library | |
---|---|---|---|---|
lightly | up-to-date | Yes | Lightly.ai | pytorch-lightning |
EasyCV | up-to-date | Yes | alibaba | mmcv |
solo-learn | 2024Jan | No | Individual | pytorch-lightning |
mmselfsup | 2023Jul | Yes | mmlab | mmcv |
vissl | archived | No | own |
結論
lightlyがオススメ。
好みでEasyCVでも良いが、mmcvは良くversion mismatchが頻発するので注意が必要。