Пакет PyTorch torchvision содержит несколько популярных встроенных наборов данных. Чтобы увидеть список встроенных наборов данных, перейдите по этой ссылке . Здесь мы покажем вам, как создать набор данных PyTorch из COCO 2017 . Скачивание набора данных COCO COCO - это крупномасштабный набор данных для обнаружения, сегментации и субтитров. Подробнее об этом можно узнать здесь . COCO 2017 имеет более 118K обучающих образцов и 5000 проверочных образцов. PyTorch torchvision..

В этом посте мы обсудим преобразование форматов аннотаций Zillin в COCO и подробно расскажем об этом процессе. Для немедленных результатов мы предоставляем готовый к использованию код Python , который позволит вам создавать аннотации COCO Object Detection из подходящих наборов данных Zillin. Чтобы узнать больше о том, как адаптировать код нашего примера к различным задачам и наборам данных, читайте дальше. Если вы читали наш предыдущий пост о создании TFRecords на основе аннотаций..