We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 093ee6c commit af46195Copy full SHA for af46195
examples/cars segmentation (camvid).ipynb
@@ -868,7 +868,7 @@
868
" gt_mask = gt_mask.squeeze()\n",
869
" \n",
870
" x_tensor = torch.from_numpy(image).to(DEVICE).unsqueeze(0)\n",
871
- " pr_mask = model.predict(x_tensor)\n",
+ " pr_mask = best_model.predict(x_tensor)\n",
872
" pr_mask = (pr_mask.squeeze().cpu().numpy().round())\n",
873
874
" visualize(\n",
0 commit comments