felipecode / coiltraine

Training framework for conditional imitation learning
MIT License
230 stars 68 forks source link

Image size #43

Open zhangdan8962 opened 2 years ago

zhangdan8962 commented 2 years ago

I noticed that the model requires images of size 88x200. I am wondering if I can input different size and modify following line accordingly or do I have to make any other modifications? https://github.com/felipecode/coiltraine/blob/29060ab5fd2ea5531686e72c621aaaca3b23f4fb/input/coil_dataset.py#L135