show_results under MultiTaskRoadExtractor() throws an error

[MariamJamilah]

Hello,

I am running the "Automatic road extraction using Deep Learning" sample notebook which uses the MultiTaskRoadExtractor() model. I get the below error when I try to execute the show_results() method to visualize the output. I have attached a screenshot for reference. Please advice

TypeError: 'AxesSubplot' object is not subscriptable

[Anonymous User]

Hi,
Can I know what batch size you used while preparing the data?

[MariamJamilah]

I used a batch size of 1

[Anonymous User]

Hmm, looks like this batch_size=1 is a problem here and needs to be fixed from our side. We will get back to you after the fix. Meanwhile, you can choose some other batch size (other than 1) and let me know if you face this issue.