test: only use color image
saparate training phase and testing phase
Fix bugs
Recode main program