#损失函数
allRank
allRank是一个基于PyTorch的框架,旨在简化神经排序学习模型的实验。它提供多种损失函数和评分函数,并支持常用评估指标如NDCG和MRR。该框架支持添加自定义损失和配置模型与训练流程,适用于研究和工业应用。同时支持GPU和CPU架构,并集成了Google云存储功能。
Active-Passive-Losses
Active-Passive-Losses项目实现了ICML 2020论文提出的归一化损失函数,用于提高深度学习模型处理噪声标签的能力。项目提供CIFAR-10和CIFAR-100数据集上的实验配置,支持多种噪声率和噪声类型。研究者可利用此代码复现结果或将新损失函数应用于自身研究。
SupContrast
SupContrast是一个开源的监督对比学习框架,致力于提升视觉表征学习效果。该项目实现了监督对比学习和SimCLR算法,在CIFAR数据集上展现出色性能。它提供简洁的损失函数实现,支持自定义数据集,并附有详细运行指南和可视化结果。在ImageNet上,SupContrast实现了79%以上的Top-1准确率。这一工具为计算机视觉领域的研究和应用提供了重要支持。
auraloss
auraloss是一个专注于音频处理的PyTorch损失函数库。它提供了丰富的时域和频域损失函数,包括ESR、SNR、STFT等,以及多种感知变换功能。该库支持多分辨率和随机分辨率STFT损失等高级特性,适用于多种音频深度学习任务。auraloss设计简洁,易于使用,并提供详细文档和示例,是音频处理研究的实用工具。
2dimageto3dmodel
该项目开发了一种新型损失函数,能够直接从单张2D图像生成3D模型,无需复杂的渲染过程。项目采用条件GAN架构实现纹理映射,并优化了点云到3D网格的转换技术。在CUB鸟类和Pascal 3D+数据集上的测试显示了显著效果。此外,项目还提供预训练模型、伪真值生成和网格生成器训练等功能,为3D重建研究领域贡献了实用工具和参考方法。