Skip to content

Latest commit

 

History

History
29 lines (16 loc) · 1.12 KB

README.md

File metadata and controls

29 lines (16 loc) · 1.12 KB

vehicle-speed-estimation

If you want to know more about this project checkout my medium post.

video

Requirements

pip3 install -r requirements.txt

How to use

Pre-trained model

You can simply use the model that I trained before. It is under models. Use the inference.ipynb to load the model and run an inference.

Train yourself

You can train the network (EfficientNet) to predict the speed of a vehicle using optical flow. If you want to train yourself, you will need to create the optical flow images first and save them as .npy files in a directory of your choice. You can do this here: SharifElfouly/opical-flow-estimation-with-RAFT.

Results

If you are interested on how well the model performs, watch this validation video on YouTube.

Another approach

You can also just stack 2 frames together so you have 6 channels for each input and feed that to a conv net. This is what I did in train_cnn_2frames.ipynb.