图像分类识别——（1）配置环境

因个人电脑配置不支持，本次项目是运行在featurize平台的云服务器上的。（官网：featurize.cn）

测试云GPU环境：GPU RTX 3060 ，显存：12GB，CPU：6 x E5-2680 v4

 

代码配置

安装库

!pip install numpy pandas matplotlib requests tqdm opencv-python pillow gc -i https://pypi.tuna.tsinghua.edu.cn/simple

 

下载Pytorch

!pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

 

 

安装mmcv -full库，作用是对视频文件进行逐帧分割保存并再串成一个视频

!pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html

 

 安装中文字体文件

!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220716-mmclassification/dataset/SimHei.ttf

 

 下载预训练模型 ImageNet 1000的类别信息，也可换成自己训练好的模型

!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220716-mmclassification/dataset/meta_data/imagenet_class_index.csv

 

 文件下载到了当前目录中，效果如下图

 

 

创建被测试图片的存储目录

import os

# 存放测试图片
os.mkdir('test_img')

# 存放结果文件
os.mkdir('output')

运行后在当前目录下创建了两个文件夹

 

 

然后从用代码下载一些图片存放到test_img目录下，也可自己找图片放在test_img目录。

# 下载测试图像文件 至 test_img 文件夹

!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220716-mmclassification/test/watermelon1.jpg -O test_img/watermelon1.jpg
!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220716-mmclassification/test/banana1.jpg -O test_img/banana1.jpg
!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220716-mmclassification/test/cat1.jpg -O test_img/cat1.jpg

# 哈士奇，来源：https://www.pexels.com/zh-cn/photo/2853130/
!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220716-mmclassification/test/husky1.jpeg -O test_img/husky1.jpeg

# 猫狗，来源：https://unsplash.com/photos/ouo1hbizWwo
!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220716-mmclassification/test/cat_dog.jpg -O test_img/cat_dog.jpg

!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220716-mmclassification/test/video_2.mp4 -O test_img/video_2.mp4

 

至此，图片分类的环境配置准备工作已完成。

 

感谢@同济子豪兄 的教程 ，主页：github.com/TommyZihao/