1.新建GitHub仓库
先在GitHub上新建一个项目用于记录
2.赛题理解
2.1理解赛题背景和赛题数据
最终达成识别天池报名的后获得的数据图片中的数字。
2.2 天池报名
1)点击比赛连接:https://tianchi.aliyun.com/competition/entrance/531795/introduction
填写个人信息后报名成功,如下图:
2)数据获取
点击赛季与数据下载对应的csv文件,里面包含所需要的赛题数据。
csv内容,依据网址下载对应的数据,用于做识别
3. 环境配置
下载Anaconda,配置环境变量,路径:D:\01-soft\20-Miniconda\Scripts,将其加入path中,创建虚拟环境:conda create -n py37_torch131 python=3.7,启用虚拟环境:activate py37_torch131,安装pytorch1.3.1:conda install pytorch=1.3.1 torchvision cudatoolkit=10.0,安装依赖库:pip install jupyter tqdm opencv-python matplotlib pandas,启用notebook:jupyter-notebook
4.分割图片
使用代码:
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
from tqdm import tqdm, tqdm_notebook
import torch
torch.manual_seed(0)
torch.backends.cudnn.deterministic = False
torch.backends.cudnn.benchmark = True
import torchvision.models as models
import torchvision.transforms as transforms
import torchvision.datasets as datasets
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
from torch.autograd import Variable
from torch.utils.data.dataset import Dataset
def parse_json(d):
arr = np.array([
d['top'], d['height'], d['left'], d['width'], d['label']
])
arr = arr.astype(int)
return arr
img = cv2.imread('../input/mchar_train/000001.png')
arr = parse_json(train_json['000001.png'])
plt.figure(figsize=(10, 10))
plt.subplot(1, arr.shape[1]+1, 1)
plt.imshow(img)
plt.xticks([]); plt.yticks([])
for idx in range(arr.shape[1]):
plt.subplot(1, arr.shape[1]+1, idx+2)
plt.imshow(img[arr[0, idx]:arr[0, idx]+arr[1, idx],arr[2, idx]:arr[2, idx]+arr[3, idx]])
plt.title(arr[4, idx])
plt.xticks([]); plt.yticks([])
效果图:
torchvision,torch安装教程网址:https://pytorch.org/,https://www.cnblogs.com/guxingy/p/11211559.html,可参考两者安装