Top Train Computer Vision Models
The models below have been fine-tuned for various train detection tasks. You can try out each model in your browser, or test an edge deployment solution (i.e. to an NVIDIA Jetson). You can use the datasets associated with the models below as a starting point for building your own train detection model.
At the bottom of this page, we have guides on how to count trains in images and videos.
by Yipli
1019 images 8 classes
1100 images 11 classes
by AL AMIN
1119 images 8 classes
by YOLOv8
158 images 16 classes
by go4av05
8659 images 40 classes
* Random Gaussian blur of between 0 and 2.5 pixels * Random brigthness adjustment of between -25 and +25 percent * Random exposure adjustment of between -25 and +25 percent * Salt and pepper noise was applied to 5 percent of pixels * annotate, and create datasets * collaborate with your team on computer vision projects * collect & organize images * export, train, and deploy computer vision models * understand and search unstructured image data * use active learning to improve your dataset over time 17 18 19 20 21 22 23 24 25 26