RoboJackets / igvc-software

The code base for the RoboNav team's IGVC robot.

https://robojackets.org/teams/robonav/

MIT License

91 stars 120 forks source link

Feat/multiclass resize #893

Closed VAM7686 closed 2 years ago

VAM7686 commented 2 years ago

Description

Added additional functionality to the multiclass segmentation for training images of different sizes

This PR does the following:

Adds more documentation to training the multiclass segmentation model on Colab
Adds support for resizing dataset to different images

Testing steps

Test Case 1

View model inferences on the shared drive when trained on 416x416 images. Results are in RoboNav > Software > model_training > full_model4_resized > predictions

Expectation: Results look correct for 416x416 images

Self Checklist

[x] I have formatted my code using make format
[x] I have tested that the new behaviour works

pep8speaks commented 2 years ago

Hello @VAM7686! Thanks for updating this PR. We checked the lines you've touched for PEP 8 issues, and found:

There are currently no PEP 8 issues detected in this Pull Request. Cheers! :beers:

Comment last updated at 2022-06-02 06:13:38 UTC