search

OlafenwaMoses / ImageAI

A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities
https://www.genxr.co/#products
MIT License
8.64k stars 2.19k forks source link

Training time on google Colab for custom model #323

Open Rakeshchandrajoshi opened 5 years ago

Rakeshchandrajoshi commented 5 years ago

Sir, Can you tell me what time it will take to train the objects on Google Colab? I have tried using your hololens and oculus but I got stuck at epoch 1/200. Can you suggest me something regarding this?

rola93 commented 5 years ago

If you can install it from sources, there was a small change to show progress more often, and even a progress bar with estimated time remaining.

Otherwise, you can modify the sources by your self. Read more about this discussion at #306

Rakeshchandrajoshi commented 5 years ago

I want to transfer learning on the yolo-coco model, and I followed your steps but unable to succeed. I thought I am somewhere wrong? Anyone who has transfer learning on Tiny-Yolo-COCO model, can suggest me the correct way via Google Colab? and also suggest me the time constraints to add a single class to that model? or someone who can share the .h5 file after transfer learning and having more classes than 80?

Rakeshchandrajoshi commented 5 years ago

If you can install it from sources, there was a small change to show progress more often, and even a progress bar with estimated time remaining.

Otherwise, you can modify the sources by your self. Read more about this discussion at #306

I want to transfer learning on the yolo-coco model, and I followed your steps but unable to succeed. I thought I am somewhere wrong? Anyone who has transfer learning on Tiny-Yolo-COCO model, can suggest me the correct way via Google Colab? and also suggest me the time constraints to add a single class to that model? or someone who can share the .h5 file after transfer learning and having more classes than 80?

rola93 commented 5 years ago

Maybe you should start by sharing some code to see what are you doing rigth now. I don't get what do you mean by didn't succed. does it raise an error?

OlafenwaMoses commented 5 years ago

@Rakeshchandrajoshi Please read the full documentation linked below to fully understand the training process.

https://github.com/OlafenwaMoses/ImageAI/blob/master/imageai/Detection/Custom/CUSTOMDETECTIONTRAINING.md