VortexNet: Neural network based on fluid dynamics
4 months ago
- #autoencoder
- #neural-networks
- #fluid-dynamics
- 代码库包含研究论文中VortexNet概念的玩具实现版本
- 展示基于偏微分方程的涡旋层及受流体力学启发的神经网络机制
- 包含MNIST和自定义图像数据集的VortexNet自编码器脚本
- 自定义图像脚本支持数据增强和潜在空间插值功能
- 需要Python 3.8+及特定依赖库如torch/torchvision/matplotlib
- MNIST数据集自动下载，自定义图像需放入'my_data/'目录
- 训练输出保存在指定'output_dir'，含日志和模型检查点
- 可使用TensorBoard监控训练过程