adamjstewart/sampler.py

Created May 5, 2022 21:41

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/adamjstewart/6416f0a80bef9d152778e94afb3e3141.js"></script>
Save adamjstewart/6416f0a80bef9d152778e94afb3e3141 to your computer and use it in GitHub Desktop.

Download ZIP

Creating a geospatial sampler with TorchGeo

Raw

	sampler = RandomGeoSampler(dataset, size=256, length=10000)
	dataloader = DataLoader(dataset, batch_size=128, sampler=sampler, collate_fn=stack_samples)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment