python猪脸识别_京东JDD猪脸识别比赛「建议收藏」

python猪脸识别_京东JDD猪脸识别比赛「建议收藏」pig_face_recognition京东JDD猪脸识别比赛pytorch-baseline1.运行环境TeslaK20c集群单节点双卡RedHat4.4.7-3Python2.7.13cuda8.0cudnn5.0pytorch0.3.02.从视频中截取出猪用yolo-9000算法,人工打label后,对ffmpeg提取出的视频帧进行猪的目标检测,框出猪的主体部分,为后续…

大家好,又见面了,我是你们的朋友全栈君。

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账号...

(0)


相关推荐

发表回复

您的电子邮箱地址不会被公开。

关注全栈程序员社区公众号