updated basic tutorials, better comments, code revision, checked it works with latest pytorch version

This commit is contained in:
Aladdin Persson
2022-12-19 23:39:48 +01:00
parent 3f53d68c4f
commit cd607c395c
14 changed files with 162 additions and 88 deletions

View File

@@ -8,7 +8,7 @@ import albumentations as A
def visualize(image):
plt.figure(figsize=(10, 10))
plt.axis('off')
plt.axis("off")
plt.imshow(image)
plt.show()
@@ -22,7 +22,7 @@ def plot_examples(images, bboxes=None):
if bboxes is not None:
img = visualize_bbox(images[i - 1], bboxes[i - 1], class_name="Elon")
else:
img = images[i-1]
img = images[i - 1]
fig.add_subplot(rows, columns, i)
plt.imshow(img)
plt.show()