A Partial Replication of MaskFormer in TensorFlow on TPUs for the TensorFlow Model Garden

arXiv:2404.18801v1 Announce Type: cross
Abstract: This paper undertakes the task of replicating the MaskFormer model a universal image segmentation model originally developed using the PyTorch framework, within the TensorFlow ecosystem, specifically optimized for execution on Tensor Processing Units (TPUs). Our implementation exploits the modular constructs available within the TensorFlow Model Garden (TFMG), encompassing elements such as the data loader, training orchestrator, and various architectural components, tailored and adapted to meet the specifications of the MaskFormer model. We address key challenges …

arxiv cs.cv cs.lg cs.se replication tensorflow tpus type