文件名称:BTH:此存储库包含CVPR 2021上介绍的BTH框架的Pytorch代码和模型
文件大小:84.23MB
文件格式:ZIP
更新时间:2024-04-02 12:02:01
Python
BTH 此存储库包含CVPR 2021上介绍的BTH框架的Pytorch代码和模型 通过双向变压器进行自我监督的视频散列 李淑艳,李秀,陆继文,周杰 使用指南 环境 火炬0.4.1 资料准备 下载功能 VGG功能由的作者友善上传。 您可以从百度云磁盘下载它们。 FCV: ://pan.baidu.com/s/1i65ccHv和YFCC: ://pan.baidu.com/s/1bqR8VCF 请在args.py中设置data_root和home_root。 您可以将这些功能放在data_root中。 预处理 在训练之前,应准备以下数据。 其中提供了一些用于FCVID的信息: 潜在功能\ bar {\ bm {h}}。 我们已将它们上传到./data/latent_feats.h5。 您也可以自己生成此文件。 您应该首先仅使用mask_loss训练BTH模型,然后在eval.
【文件预览】:
BTH-main
----.DS_Store(6KB)
----README.md(2KB)
----BTH()
--------results()
--------optim_schedule.py(1KB)
--------data()
--------model()
--------eval.py(4KB)
--------models()
--------utils()
--------README.md(2KB)
--------tools.py(3KB)
--------data.py(5KB)
--------train.py(4KB)