这个仓库专注于最优传输(Optimal Transport)理论的研究、实现和应用。
optimal_transport/
├── docs/ # 研究资料和文档
│ ├── papers/ # 相关论文
│ ├── notes/ # 学习笔记
│ └── references/ # 参考资料
├── src/ # 源代码
│ ├── algorithms/ # 算法实现
│ ├── utils/ # 工具函数
│ └── models/ # 模型定义
├── experiments/ # 实验代码
│ ├── training/ # 训练脚本
│ └── evaluation/ # 评估脚本
├── tests/ # 测试代码
├── data/ # 数据集
├── results/ # 实验结果
└── notebooks/ # Jupyter notebooks
- 最优传输理论基础
- Wasserstein距离计算
- Sinkhorn算法实现
- 生成模型中的应用
- 图像处理应用
- Python 3.8+
- PyTorch
- NumPy
- SciPy
- Matplotlib
git clone https://github.com/yourusername/optimal_transport.git
cd optimal_transport
pip install -r requirements.txt# 基本使用示例将在后续添加欢迎提交Issue和Pull Request来改进这个项目。
MIT License