Description - Vision based methods using deep learning such as CNN to perform terrain recognition (sandy/rocky/grass/marshy) enhanced with implicit quantities information such as the roughness, slipperiness, an important aspect for high-level environment perception.
- Deployed ML model link: https://terrain.streamlit.app/
- Numpy
- Pandas
- Matplotlib
- Albumantations
- Torch
- Torchvision
- PIL
- Pickle
- Streamlite
ResNet18 (pre-trained)with oneFully Connected Layer(512 x 4) predicting4 classes
- 31517
trainingimages with4 classesnamelyGrassy,Marshy,Rocky,Sandy - 6765
validationimages and 6769testimages - Normalized with the pretrained ResNet18 measures.
- The pre-trained layers are
freezedand only the lastFC layerwas trainined for 1 complete pass through all the samples. Mini_batch size: 32
- Accuracy on
validation_set91.1 % and ontest_set93.8% Mini_batch size: 32
- The model is deployed using
Streamlitlibrary - we
Pickeledthe model for its interation with Streamlit's "app.py" Image uploadercoloumn that uploads image to model and recieves predicted index as return value- The result is displayed on the website.
- Manav Jain (Leader): https://github.com/manavjain2005
- Harsh Singh : https://github.com/hharshas
- Abhinav Jain : https://github.com/jainjabhi05
- Aadhya Jain : https://github.com/aadhya0002
- Dyuti Ballav Paul : https://github.com/dyuti01
- Pratham Todi : https://github.com/pra1ham28