老师视频课
一、Python开发环境安装
1、安装anaconda,官网打开比较慢,可以搜索 清华大学开源软件镜像站 如下图所示:
2、安装社区版pycharm
3、cmd中创建虚拟环境 (ctrl+C结束正在运行的程序)
二、第一个深度学习实例:手写字符识别
代码:https://github.com/mivlab/AI_course
创建虚拟环境:
conda env list
(查看有什么环境)
conda create -n AI python=3.8.5
(创建环境AI,Python版本为3.8.5)
conda activate AI
(激活环境)
deactivate
(退出虚拟环境)
pip3 install torch torchvision torchaudio
(下载Pytorch,这样下载的是cpu最新版本的,cuda版本的去官网下)
pip install opencv-python
(下载Opencv)
运行后显示:
因为需要输入一个命令行参数,如下图点击Edit Configuration
然后把训练数据集的绝对路径(--datapath=E:\WeiYingYing\AI_course-master\AI_course-master\classify_pytorch\images
)填到参数一栏:
然后点右键运行即可训练,训练完成后运行tes_mnist.py
三、一些操作方法
1、调试代码
打上断点 然后运行debug可以看到类型和一些信息
当某个函数用法和含义不熟悉时可以搜索引擎搜索pytorch官网,在PyTorch的主页上点docs打开官方文档,搜索max,如下图即可看到函数的用法和例子:
2、查找引用的自定义函数所在位置及内容(选中函数->右键->Go To->Type Declaration)
3、性别识别案例 train_sex.py
import torch
import math
import torch.nn as nn
from torch.autograd import Variable
from torchvision import transforms, models
import argparse
import os
from torch.utils.data import DataLoader
from torch.utils.data import Dataset
from dataloader import mnist_loader as ml
from models.cnn import Net
from toonnx import to_onnx
class SexDataset(Dataset):
def __init__(self, txt, transform=None, target_transform=None):
fh = open(txt, 'r')
imgs = []
for line in fh:
line = line.strip('\n')#通过strip把一些换行符和空格去掉
line = line.rstrip(