大家好，又見面了，我是你們的朋友全棧君。
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.提交結果
轉換分數格式為所需格式並提交結果。
發布者：全棧程序員棧長，轉載請注明出處：https://javaforall.cn/151872.html原文鏈接：https://javaforall.cn
