search

ZJCV / RotNet

Image rotation correction based on DeepLearning
Apache License 2.0
21 stars 7 forks source link

Train model #1

Closed thienbui closed 3 years ago

thienbui commented 3 years ago

How to create dataset to train my model? Thanks

zjykzj commented 3 years ago

How to create dataset to train my model? Thanks

you can refer to RotNet/rotnet/data/datasets/cifar.py to create custom dataset; then register it to RotNet/rotnet/data/datasets/build.py; finally use configs/xxx.yaml to train

Or wait for a while and I will create a more convenient way to train custom data.

thienbui commented 3 years ago

Thanks, you can upload some samples of dataset ? Structure of folder data train. Thanks so much!

zjykzj commented 3 years ago

Thanks, you can upload some samples of dataset ? Structure of folder data train. Thanks so much!

Hi, I update the README. You can check how to add custom dataset

thienbui commented 3 years ago

Thanks so much!