Skip to content

earhian/Humpback-Whale-Identification-1st-

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

34 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Kaggle Humpback Whale Identification Challenge 1st place code

Recent Updates

[2019.3.1 16:44] uploading mask file

Dependencies

  • python==3.6
  • torch==0.4.1
  • torchvision==0.2.1

Solution

https://www.kaggle.com/c/humpback-whale-identification/discussion/82366

prepare data

competition dataset

trainset -> ./input/train
testset -> ./input/test

mask

cd input
unzip mask.zip
download model_50A_slim_ensemble.csv(https://drive.google.com/file/d/1hfOu3_JR0vWJkNlRhKwhqJDaF3ID2vRs/view?usp=sharing) into ./input

playground data

download playground data, then put them into input/train
https://www.kaggle.com/c/whale-categorization-playground/data

Train

line 301 in train.py
step 1.
       freeze = False
       model_name = 'senet154'
       min_num_class = 10
       checkPoint_start = 0
       lr = 3e-4
       #until train map5 >= 0.98

step 2.
       freeze = True
       model_name = 'senet154'
       min_num_class = 0
       checkPoint_start = best checkPoint of step 1
       lr = 3e-4

step 3.
       freeze = True
       model_name = 'senet154'
       min_num_class = 0
       checkPoint_start = best checkPoint of step 2
       lr = 3e-5

Test

line 99 in test.py
       checkPoint_start = best checkPoint of step 3