C-3-Framework
C-3-Framework copied to clipboard
准备开源代码。
由于临时出差取消,近两天我将完成代码的发布。基本框架不变,主要关注于bug的修复和各个trainer之间的部分代码同步。
后续暂定任务为:
- [x] 加入CMTL的完整训练代码和部分实验结果。感谢@SelinaFelton的贡献。(4.20前完成)【已完成】
- [x] 替换预训练网络的预训练模型加载代码,使用PyTorch官方推荐代码加载。(5.1前完成)【已完成】
- [ ] 撰写一份简单的技术报告(technical reports),简要介绍框架特性以及使用方式。(5.30前完成)【中文版正在进行】
其他任务后续在此issue中一次增加。
- [ ] 部分数据路径的一致性修改。
- [ ] 是否替换csv或者mat的gt存储方式,改为pgm, tiff等图像格式以增加读取速度。
- [ ] 是否提供数据预存储到内存中,以增加训练速度。
- [x] 增加断点训练功能,增加optimizer状态保存及加载。 (目前完成trainer和trainer_for_M2TCC的断点训练。其他trainer在完成其他相应模型后更新。)
感谢大佬