**** Thank you for sharing this code. But it will be great if there is an example code to train a new dataset.