show_results under MultiTaskRoadExtractor() throws an error

Question

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

Error.PNG

Anonymous User · Answer

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.

MariamJamilah · Answer

I used a batch size of 1

Anonymous User · Answer

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