PyTorch Vision (TorchVision)

TorchVision é um pacote no PyTorch projetado para facilitar o processo de desenvolvimento de aplicativos de visão computacional. Ele oferece uma coleção de conjuntos de dados populares, como ImageNet e COCO, juntamente com uma variedade de modelos pré-treinados que podem ser facilmente integrados a projetos. Transformações para pré-processamento e aumento de imagens também estão incluídas, agilizando a preparação de dados para o treinamento de modelos de aprendizado profundo. Ao fornecer esses recursos, o TorchVision permite que os desenvolvedores se concentrem na arquitetura e treinamento do modelo sem precisar criar cada componente do zero.