-
Notifications
You must be signed in to change notification settings - Fork 4
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add Progress Bar #80
Comments
Thanks for the issue @work4cs. When you pass a list of images to the predict() function these images are processed as a group. If you are using a "cuda" or "mps" device the images will be loaded onto the GPU and processed in a single encode_image operation. Typically this method will be faster than processing the images one at a time. @work4cs Which device are you using "cpu", "cuda", or "mps"(default is "cpu")? How many images are you passing to the |
Oh it seems like we are dealing with a single batch now. If a batch of images can fit in GPU or CPU, then we don't need to monitor the time and progress since it can finish in the blink of an eye. Would you like to extend it to multiple batch with a |
I think a |
Sounds good! |
Adds --batch-size for predict command and classifier predict() methods. Fixes #80
Adds --batch-size for predict command and classifier predict() methods. Fixes #80
Adds --batch-size for predict command and classifier predict() methods. Fixes #80
Adds --batch-size for predict command. Adds batch_size to classifier predict() function. Default batch size is 10. Updates early notebooks to process images in a batch. Fixes #80
Could we add progress bar (such as
tqdm
) when we add a list of images toTreeOfLifeClassifier
orCustomLabelsClassifier
?The text was updated successfully, but these errors were encountered: