CSIP

安装RainbowNeko Engine.

修改cfgs/py/train/contrastive/csip.py中数据和训练相关配置。

训练

# 多卡
neko_train --cfg cfgs/py/train/contrastive/csip.py
# 单卡
neko_train_1gpu --cfg cfgs/py/train/contrastive/csip.py

换InfoNCE Loss:

neko_train --cfg cfgs/py/train/contrastive/csip_info_nce.py

推理

python csip_flow.py

输出两两图片的相似度