-
Notifications
You must be signed in to change notification settings - Fork 174
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
Training my own dataset #36
Comments
+1 |
the same question |
Here's the solution that worked for me. I created a csv file containing path for image and its label.
I changed bit_hyperrule.py accordingly.
You can change the resolution from the get_resolution function (from bit.hyperrule.py) if needed. I changed /bit_pytorch/train.py file to load the custom data.
Keep customDataLoader.csv as well as customDataset.py in the root folder (with bit_hyperrule.py). Run the code using command: I had changed the default values (for batch_size, etc.) from the code itself. Hope that helps. ! References: |
I have tried yours,but it raise:AttributeError: 'MyDataset' object has no attribute 'classes'. |
When I tried the mentioned solution, it worked.
|
OK,Thanks |
Hi
what would be the data format and folder layout that I need to setup if I have my own dataset?
Thx
The text was updated successfully, but these errors were encountered: