大家好,又见面了,我是你们的朋友全栈君。
pig_face_recognition
京东JDD猪脸识别比赛
pytorch-baseline
1.运行环境
TeslaK20c集群单节点双卡 Red Hat 4.4.7-3 Python 2.7.13 cuda 8.0 cudnn 5.0 pytorch 0.3.0
2.从视频中截取出猪
用yolo-9000算法,人工打label后,对ffmpeg提取出的视频帧进行猪的目标检测,框出猪的主体部分,为后续分类做基础。
3.数据预处理
把下载的数据集预处理,生成torchvision.datasets.ImageFolder接口需要的文件夹格式,并使用torchvision.transforms中的方法进行数据增强,具体采用了RandomResizedCrop、RandomHorizontalFlip、ColorJitter以及Normalize.
4.train from scratch or fitune from imagenet
尝试多种resnet和densenet网络,最后选择较好的结果进行平均融合。
5.测试结果
对测试数据集进行前向运算得到预测分类概率并保存。
6.提交结果
转换分数格式为所需格式并提交结果。
发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/151872.html原文链接:https://javaforall.cn
【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛
【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...