Kunwar Raj Singh 5d3310ce56 MaskRCNN Inference (#884) ... * MaskRCNN weights loading * backbone maybe works * backbone works, but resnet body atol 1e-3 * RPN Call, but veryy wrong output * fixed topk * RPN maybe works, not sure about nms * Fix cursed modules * add back editorconfig * Full call, wrong output * Full call works * fix mask * use NMS from retinanet * Removing extra funcs * refactor * readable * Add example to run model * remove filter * Fix split, batched inference is worse * Fix image sizes * Matching reference * merge master * add filter on top detections * cuda backend fixed * add model eval and spec * convert images to rgb * fix eval * simplify examples code * remove extra code * meshgrid using tinygrad * removing numpy * roi align, floor, ceil * remove numpy from level_mapper * remove numpy from pooler * Revert "Merge branch 'master' of github.com:kunwar31/tinygrad into mrcnn-inference" This reverts commit 4b95a3cb49, reversing changes made to 98f2b1fa2e. * roi align gather * fix master merge * revert to old floor, ceil as ints present in domain * use log2 op * fix indexes * weird bug with ints and gpu * weird bug with ints and gpu * refactors, add env var for gather * floor with contiguous, where * refactor topk, sort * remove staticmethod * refactor stride * remove log2 mlop * realize -> contiguous * refactor forward * remove num_classes, stride_in_1x1 from state * refactor forward * refactoring * flake8 * removing numpy in anchor gen, use numpy for gather, nonzero, optimize topk * keep using tinygrad for smaller gathers * fix empty tensors * comms * move from tensor.py * resnet test passing * add coco dataset back * fix spaces * add test for log2 * no need to create Tensors * no need to create Tensors --------- Co-authored-by: Kunwar Raj Singh <kunwar31@pop-os.localdomain>		2023-06-25 15:37:51 -07:00
..
mnist	move datasets into datasets	2021-10-30 19:55:50 -07:00
__init__.py	start on mlperf models	2023-05-10 16:30:49 -07:00
coco.py	MaskRCNN Inference (#884)	2023-06-25 15:37:51 -07:00
imagenet.py	Cleanup mlperf (#797)	2023-05-25 11:36:43 -07:00
imagenet_download.py	imagenet download and prepare (#928)	2023-06-08 12:55:33 -07:00
kits19.py	Add MLPerf UNet3D model (#775)	2023-05-28 20:38:19 -07:00
librispeech.py	Cleanup mlperf (#797)	2023-05-25 11:36:43 -07:00
openimages.py	add retinanet with resnet backbone (#813)	2023-05-28 20:20:16 -07:00
preprocess_imagenet.py	int8/uint8 support (#837)	2023-05-28 23:15:06 -07:00
squad.py	Add mlperf bert model (#803)	2023-05-27 14:53:32 -07:00