机器学习
文章平均质量分 62
二代红鱼
一切问题都是历史遗留问题
展开
-
(关键点检测)YOLOv8实现多类人体姿态估计的输出格式分析
(关键点检测)YOLOv8实现多类人体姿态估计的输出格式分析原创 2024-03-07 11:47:19 · 529 阅读 · 0 评论 -
PyTorch模型保存与使用
PyTorch模型保存与使用模型的保存模型的使用PyTorch模型保存与使用训练好的模型保存下来,这样在需要的时候就可以直接调用。模型的保存保存模型有两种方法,一种是保存模型的参数,再次加载时需要原有模型的信息。 另一种是保存模型的所有信息,可以直接加载为原有模型。需要注意的是,如果原有模型是在GPU上训练的,那么在加载到只有CPU 的电脑上时,需要在 torch.load() 时,加上 map_location=torch.device(‘cpu’) 参数变换模型参数到CPU .原创 2021-09-24 23:04:25 · 4750 阅读 · 0 评论 -
MNIST 数据集下载及图片可视化
MNIST 数据集下载及可视化MNIST 介绍下载数据集并保存成图片MNIST 数据集下载及可视化最近用到了 MNIST 训练神经网络,但是下载下来的数据集默认是 `*-ubyte.gz` 的压缩包。 无法直观的看到数据本身的图片是什么样子,训练的神经网络达到99% 的正确率还是觉的一头雾水, 想看一下自己到底用的什么图片。但是网上关于 MNIST数据集可视化 的文章有些比较繁琐,特在此记录将MNIST数据集下载下来,并以图片形式保存到本地的操作步骤。MNIST 介绍MNIST .原创 2021-09-19 17:23:37 · 4373 阅读 · 4 评论