/Distracted_Driver_Detection_NIN-GAM

Primary LanguageHTML

依赖

  • tensorflow 1.0.1
  • numpy
  • matplotlib
  • pandas
  • PIL(pillow)
  • sklearn

运行时间

  • vgg_gam训练2 epoch,10 KFold,batchsize 20,时间约6个小时,生成csv预测结果约1小时
  • nin训练3 epoch,10 KFold,batchsize 20,时间约8小时,生成csv预测结果约0.5小时
  • deep_nin训练2 epoch,10 KFold,batchsize 10,时间约13小时,csv预测结果约0.5小时

运行硬件

  • CPU i7 6770
  • RAM 16G
  • GPU GTX1080 8G

checkpoint都上传到了百度网盘