YCmove

Happy Programmer

fmassa / query_distribution.py

Created June 28, 2020 16:11

Code to reproduce Fig 7 in "End to End Object Detection with Transformers"

	# this file needs to be added to the root folder of detr github repo
	import torch
	import time
	import torchvision

	import numpy as np
	import tqdm

	import matplotlib.pyplot as plt