Skip to content

Pytorch-Wildlife v1.0.1

Compare
Choose a tag to compare
@zhmiao zhmiao released this 29 Dec 04:43
· 2115 commits to main since this release

In this updated version, we've addressed some memory leak problems that originated from utilizing numpy within Pytorch's dataset and dataloader. The primary cause was the letterbox function, which we've now completely overhauled. Initially, this function, implemented by Ultralytics, relied entirely on numpy. However, numpy is known to conflict with Pytorch's dataset and dataloader, as documented in this issue (pytorch/pytorch#13246). To circumvent this problem, we've re-implemented the function using only PIL and Pytorch.

Alongside this, we've also modified some post-processing functions to prepare our codebase for dockerization. Currently, we're in the process of launching a Docker image with Pytorch-Wildlife to simplify environment management. We appreciate your patience during this time!