Если вы хотите, чтобы что-то обрабатывало более сложные детали, вы можете настроить OpenGL с помощью pygame, тогда используйте
surface = pygame.image.load(fileobj)
Где fileobj - это любой объект с файловым интерфейсом,Вам нужно изменить генератор PNG для записи в этот объект, а не в реальный файл.