Training on other dataset

[pomelyu]

Thanks for the nice work
The result from the test script is great.

However, how could I train the network on my own dataset.
I could not find the code for training the PCA and Edge network.
I guess these two parts are highly related to the training data, hence could not use the pretrained model in the repository.

Thanks