Thanks! Custom dataset indeed on the todo list. The feature is half done already, I should be able to release it soon.