Skip to content

Instantly share code, notes, and snippets.

@Yuktha-Majella

Yuktha-Majella/Predication

Last active June 10, 2021 18:07
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Yuktha-Majella/dbdfc003abc48da9b3a71be92c8ff40b to your computer and use it in GitHub Desktop.
Save Yuktha-Majella/dbdfc003abc48da9b3a71be92c8ff40b to your computer and use it in GitHub Desktop.
Download ZIP
Prediction using the trained model
Raw
Predication
test_data = ImageList.from_df(test, cols=['image_names'], path='/content/drive/MyDrive/CV_Vehicle_classification/train_data/images')
t_rn50 = load_learner('/content/drive/MyDrive/CV_Vehicle_classification/model/', 'Bmodel_fastai_resnet50.h5', test = test_data)
y_trn50 = t_rn50.TTA(ds_type = DatasetType.Test)
preds = y_trn50[0].argmax(-1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment