Lesson 1 In-Class Discussion ✅

[dreambeats (JamesT)]

How to add test data to ImageDataBunch and predict labels for test data

After taking lesson 1, I tried out image classification with stanford cars dataset. I was having trouble with using the test set in ImageDataBunch. This is how my ImageDataBunch looks now. data = ImageDataBunch.from_df(path/'car_ims', train_df, ds_tfms=get_transforms(), size=224, bs=bs).normalize(imagenet_stats) test_data = ImageList.from_df(test_df, path/'car_ims') data.add_test(test_data) ImageDataBunch; Train: LabelList (6516 items) x: ImageList Image (3, 224, 2…

I responded here with a reasonable way to do it. If you already have a test set within your data bunch skip the steps that add a test set to it and run the predictions. You will have to take an argmax from the logits.