/maj-detect

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

麻雀牌面识别检测模型

概述

以雀魂游戏的麻雀牌面作为学习数据,训练出可以检测并识别牌面的深度学习模型。模型采用YOLO建模,各指标如后文所示。

指标

硬件指标

  • 显卡: GeForce GTX 1060
  • 显卡驱动:531.79
  • CUDA版本:11.8

软件指标

  • Torch版本:2.0.1+cu118
  • TorchAudio版本:2.0.2+cu118
  • TorchVision版本:0.15.2+cu118