文件名称：Siamese-Neural-Networks:使用MNIST数据集在PyTorch中实现暹罗神经网络
文件大小：189KB
文件格式：ZIP
更新时间：2024-06-07 10:26:44
Python
连体神经网络的单次图像识别 PyTorch中的的实现，在数据集上进行了培训和测试。 要求 火炬视觉== 0.5.0 火炬== 1.4.0 numpy == 1.16.3 pytorch_lightning == 0.5.3.2 枕头== 7.0.0 提供requirements.txt 指示 该项目使用 ，它是上的轻量级包装。 该项目遵循格式。 只需运行cpu_run.py或gpu_run.py即可下载MNIST数据集并开始训练。 结果 保持测试集上的最高10次单发准确性为97.5％，可与监督分类模型相提并论。 手动选择支持集。 参考 暹罗火炬
【文件预览】：
Siamese-Neural-Networks-master
----gpu_run.py(2KB)
----val_acc.png(67KB)
----cpu_run.py(1KB)
----train_loss.png(139KB)
----requirements.txt(87B)
----model.py(8KB)
----README.md(2KB)
----data.py(6KB)