mirror of
https://github.com/prise6/smart-iss-posts
synced 2024-05-17 21:06:33 +02:00
33 lines
661 B
Python
33 lines
661 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
||
|
import PIL
|
||
|
import os
|
||
|
|
||
|
class Tools:
|
||
|
|
||
|
@staticmethod
|
||
|
def display_one_picture(array):
|
||
|
array = array.astype('uint8')
|
||
|
return PIL.Image.fromarray(array, 'RGB')
|
||
|
|
||
|
@staticmethod
|
||
|
def display_one_picture_scaled(array):
|
||
|
array = array * 255
|
||
|
return Tools.display_one_picture(array)
|
||
|
|
||
|
@staticmethod
|
||
|
def display_index_picture_scaled(array, index = 0):
|
||
|
return Tools.display_one_picture_scaled(array[index])
|
||
|
|
||
|
@staticmethod
|
||
|
def display_index_picture(array, index = 0):
|
||
|
return Tools.display_one_picture(array[index])
|
||
|
|
||
|
@staticmethod
|
||
|
def create_dir_if_not_exists(path):
|
||
|
if not os.path.exists(path):
|
||
|
os.makedirs(path)
|
||
|
return path
|
||
|
|
||
|
|