Conversation
|
Thanks a lot for your contribution. However, if I understand correctly your code, there is one case that is not covered. In my Dataset class, there are two options whether to load everything to memory or just load it when you need it and then it provides a path instead of a numpy array. In the first case, your code would work. In the second case that is implemented using this: in my original code it will not. If you could incorporate this possibility in your pull request, it would be great! |
|
We'll do another pass over it. Anyway, the preprocessing is done inefficiently. |
|
@gyglim I agree that the preprocessing is inefficient and should be move to a more TF friendly manner, but I haven't had the time to do that. |
|
I think also that the new TF dataset api could help to have a more clean solution |
I have moved
preprocess_imgtopreprocess_img_tfwhich does the preprocessing in TF. If you are happy with that we can remove the oldpreprocess_imgmethod.Let me know what you think!
cc @gyglim