The RPi HQ camera is a nice step up from the regular RPi camera while being not too expensive too. Incidentally, I also have a project using that [0] but unfortunately no trains where I live.

[0] https://daniel.lawrence.lu/blog/y2022m01d27/