Skip to content

Instantly share code, notes, and snippets.

@FrancescoSaverioZuppichini

FrancescoSaverioZuppichini/non-max-suppression-in-pytorch-1.py

Created October 26, 2022 07:56
Show Gist options
  • Save FrancescoSaverioZuppichini/3499d5692f0f097853492435d63a18d2 to your computer and use it in GitHub Desktop.
Save FrancescoSaverioZuppichini/3499d5692f0f097853492435d63a18d2 to your computer and use it in GitHub Desktop.
Download ZIP
Test
Raw
non-max-suppression-in-pytorch-1.py
from PIL import Image
import torch
import matplotlib.pyplot as plt
import numpy as np
# credit https://i0.wp.com/craffic.co.in/wp-content/uploads/2021/02/ai-remastered-rick-astley-never-gonna-give-you-up.jpg?w=1600&ssl=1
img = Image.open("./samples/never-gonna-give-you-up.webp")
img
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment