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が頻発するので注意が必要。