随笔分类 - 机器学习
摘要：pytorch猫狗识别实现 不要轻视任何一个环节 写在前面：看官方文档很重要！ 数据集 制作数据集 按照下图整理数据集文件结构，cat里面只放猫图、dog里面只放狗图 读取数据集 这里也可以通过重写from torch.utils.data import Dataset里面的Dataset类，自定义
阅读全文
摘要：solo2试跑报错记录 跑自己训练集前，注意自己的gpu个数和选用的训练文件对应！ 1.solov2训练自己数据集 1.1先跑通demo demo跑通说明环境配置没有问题 出现UserWarning: This overload of nonzero is deprecated，说torch的non
阅读全文
摘要：win10下solov2环境搭建 我的环境：NVIDIA-SMI：496.49、VS社区版2019、anaconda3 官方给了mmdetetion和detetron2两个版本，我配的是mmdetetion python版本3.7 为了让环境有序，创建一个虚拟环境 conda create -n t
阅读全文
