RetinaNet dataloader (#9442)

* retinanet dataloader

* remove batch_size from generate_anchors

* refactor kits19 dataset tests

* add tests for dataloader

* fix testing setup and cleanups

* remove unused import
This commit is contained in:
Francis Lata
2025-03-21 13:36:41 -04:00
committed by GitHub
parent 58206fa8a9
commit eb95825eea
13 changed files with 964 additions and 16 deletions

View File

@@ -68,7 +68,10 @@ setup(name='tinygrad',
"nibabel",
"bottle",
"ggml-python",
"capstone"
"capstone",
"pycocotools",
"boto3",
"pandas"
],
'docs': [
"mkdocs",