程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
您现在的位置: 程式師世界 >> 編程語言 >  >> 更多編程語言 >> Python

Python pig face recognition_ JD JDD pig face recognition competition suggestions collection

編輯:Python

Hello everyone , I meet you again , I'm your friend, Quan Jun .

pig_face_recognition

JD.COM JDD Pig face recognition competition

pytorch-baseline

1. Running environment

TeslaK20c Cluster single node dual card Red Hat 4.4.7-3 Python 2.7.13 cuda 8.0 cudnn 5.0 pytorch 0.3.0

2. Take the pig from the video

use yolo-9000 Algorithm , Manual beating label after , Yes ffmpeg The extracted video frame is used for pig target detection , Frame the main part of the pig , Make a foundation for subsequent classification .

3. Data preprocessing

Preprocess the downloaded data set , Generate torchvision.datasets.ImageFolder The folder format required by the interface , And use torchvision.transforms Data enhancement using the method in , Specifically used RandomResizedCrop、RandomHorizontalFlip、ColorJitter as well as Normalize.

4.train from scratch or fitune from imagenet

Try a variety of resnet and densenet The Internet , Finally, better results are selected for average fusion .

5. test result

Carry out forward operation on the test data set to obtain the predicted classification probability and save it .

6. Submit results

Convert score format to desired format and submit results .

Publisher : Full stack programmer stack length , Reprint please indicate the source :https://javaforall.cn/151872.html Link to the original text :https://javaforall.cn


  1. 上一篇文章:
  2. 下一篇文章:
Copyright © 程式師世界 All Rights Reserved