上传者: 42109125
|
上传时间: 2022-03-07 09:47:33
|
文件大小: 189KB
|
文件类型: -
连体神经网络的单次图像识别
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%,可与监督分类模型相提并论。 手动选择支持集。
参考
暹罗火炬