Fixed bug that prevented to use center crop

real-stanford / diffusion_policy

[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion

https://diffusion-policy.cs.columbia.edu/

MIT License

1.09k stars 203 forks source link

Fixed bug that prevented to use center crop #66

Open chisarie opened 2 months ago

chisarie commented 2 months ago

Hello, first of all, congratulation on this amazing work, it is really inspiring!

I am building on top of this repo for my next project, and I noticed a small bug that prevents to use CenterCrop when setting random_crop: False in the .yaml config file. The variable name was wrong, so it was getting overwritten a couple of lines later.

This pull request should fix it, hope it helps!