gy15635282365
/
tensorlayer3
Not watched
1
0
Fork 0
Code
Releases 0 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 0e8a2ae701
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '0e8a2ae701'
${ noResults }
tensorlayer3/tests/dataflow/test_dataflow_image.py

280 lines
10 kB
Raw Blame History 

  1. #!/usr/bin/env python

  2. # -*- coding: utf-8 -*-

  3. 

  4. import os

  5. import unittest

  6. 

  7. os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'

  8. 

  9. import tensorlayer as tl

  10. 

  11. from tests.utils import CustomTestCase

  12. 

  13. 

  14. class Dataflow_Image_Test(CustomTestCase):

  15. 

  16.     @classmethod

  17.     def setUpClass(self):

  18.         self.input_shape = [1, 100, 100, 3]

  19.         self.input_layer = tl.layers.Input(self.input_shape, name='input_layer')

  20.         self.input_shape_1 = [100, 100, 3]

  21.         self.input_layer_1 = tl.layers.Input(self.input_shape_1, name='input_layer_1')

  22. 

  23.         self.centralcrop_1 = tl.dataflow.image.CentralCrop(self.input_layer, central_fraction=0.5)

  24.         self.centralcrop_2 = tl.dataflow.image.CentralCrop(self.input_layer, size=60)

  25. 

  26.         self.hsvtorgb = tl.dataflow.image.HsvToRgb(self.input_layer)

  27. 

  28.         self.adjustbrightness = tl.dataflow.image.AdjustBrightness(self.input_layer, factor=0.5)

  29.         self.adjustconstrast = tl.dataflow.image.AdjustContrast(self.input_layer, factor=0.5)

  30.         self.adjusthue = tl.dataflow.image.AdjustHue(self.input_layer, factor=0.5)

  31.         self.adjustsaturation = tl.dataflow.image.AdjustSaturation(self.input_layer, factor=0.5)

  32. 

  33.         self.crop = tl.dataflow.image.Crop(

  34.             self.input_layer, offset_height=20, offset_width=20, target_height=60, target_width=60

  35.         )

  36. 

  37.         self.fliphorizontal = tl.dataflow.image.FlipHorizontal(self.input_layer)

  38.         self.flipvertical = tl.dataflow.image.FlipVertical(self.input_layer)

  39. 

  40.         self.rgbtogray = tl.dataflow.image.RgbToGray(self.input_layer)

  41.         self.graytorgb = tl.dataflow.image.GrayToRgb(self.rgbtogray)

  42. 

  43.         self.padtoboundingbox = tl.dataflow.image.PadToBoundingbox(

  44.             self.input_layer, offset_height=20, offset_width=20, target_height=150, target_width=150

  45.         )

  46. 

  47.         self.pad_1 = tl.dataflow.image.Pad(self.input_layer, padding=10, padding_value=1, mode='constant')

  48.         self.pad_2 = tl.dataflow.image.Pad(self.input_layer, padding=(10, 10), mode='REFLECT')

  49.         self.pad_3 = tl.dataflow.image.Pad(self.input_layer, padding=(10, 20, 30, 40), mode='SYMMETRIC')

  50. 

  51.         self.standardization_1 = tl.dataflow.image.Standardization(

  52.             self.input_layer, mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)

  53.         )

  54.         self.standardization_2 = tl.dataflow.image.Standardization(self.input_layer, channel_mode=False)

  55.         self.standardization_3 = tl.dataflow.image.Standardization(self.input_layer, channel_mode=True)

  56. 

  57.         self.randombrightness = tl.dataflow.image.RandomBrightness(self.input_layer, factor=0.5)

  58.         self.randomcontrast = tl.dataflow.image.RandomContrast(self.input_layer, lower=0.2, upper=0.5)

  59.         self.randomhue = tl.dataflow.image.RandomHue(self.input_layer, factor=0.5)

  60.         self.randomsaturation = tl.dataflow.image.RandomSaturation(self.input_layer, lower=0.2, upper=0.5)

  61. 

  62.         self.randomcrop_1 = tl.dataflow.image.RandomCrop(self.input_layer, size=50)

  63.         self.randomcrop_2 = tl.dataflow.image.RandomCrop(self.input_layer, size=(50, 60))

  64. 

  65.         self.resize_1 = tl.dataflow.image.Resize(

  66.             self.input_layer, size=46, method='bilinear', preserve_aspect_ratio=False, antialias=True

  67.         )

  68. 

  69.         self.resize_2 = tl.dataflow.image.Resize(

  70.             self.input_layer, size=(32, 45), method='bilinear', preserve_aspect_ratio=True, antialias=False

  71.         )

  72. 

  73.         self.croporpad = tl.dataflow.image.CropOrPad(self.input_layer, target_height=50, target_width=150)

  74.         self.resizeandpad = tl.dataflow.image.ResizeAndPad(

  75.             self.input_layer, target_height=50, target_width=150, method='bilinear'

  76.         )

  77.         self.rgbtohsv = tl.dataflow.image.RgbToHsv(self.input_layer)

  78.         self.transpose = tl.dataflow.image.Transpose(self.input_layer, order=(3, 2, 1, 0))

  79.         self.randomrotation = tl.dataflow.image.RandomRotation(

  80.             self.input_layer_1, degrees=60, fill_mode='nearest', fill_value=1

  81.         )

  82.         self.randomshift_1 = tl.dataflow.image.RandomShift(

  83.             self.input_layer_1, shift=0.5, fill_mode='nearest', fill_value=0

  84.         )

  85.         self.randomshift_2 = tl.dataflow.image.RandomShift(

  86.             self.input_layer_1, shift=(0.5, 0.4), fill_mode='nearest', fill_value=0

  87.         )

  88. 

  89.         self.randomshear = tl.dataflow.image.RandomShear(

  90.             self.input_layer_1, degree=30, fill_mode='nearest', fill_value=1

  91.         )

  92. 

  93.         self.randomzoom_1 = tl.dataflow.image.RandomZoom(

  94.             self.input_layer_1, zoom_range=0.5, fill_mode='nearest', fill_value=1

  95.         )

  96.         self.randomzoom_2 = tl.dataflow.image.RandomZoom(

  97.             self.input_layer_1, zoom_range=(0.5, 0.4), fill_mode='nearest', fill_value=1

  98.         )

  99. 

  100.         self.rescale = tl.dataflow.image.Rescale(self.input_layer, scale=3, offset=4)

  101.         self.randomflipvertical = tl.dataflow.image.RandomFlipVertical(self.input_layer)

  102.         self.randomfliphorizontal = tl.dataflow.image.RandomFlipHorizontal(self.input_layer)

  103.         self.hwc2chw = tl.dataflow.image.HWC2CHW(self.input_layer)

  104.         self.chw2hwc = tl.dataflow.image.CHW2HWC(self.hwc2chw)

  105. 

  106.     @classmethod

  107.     def tearDownClass(self):

  108.         pass

  109. 

  110.     def test_centralcrop_1(self):

  111. 

  112.         self.assertEqual(tl.get_tensor_shape(self.centralcrop_1), [1, 50, 50, 3])

  113. 

  114.     def test_centralcrop_2(self):

  115. 

  116.         self.assertEqual(tl.get_tensor_shape(self.centralcrop_2), [1, 60, 60, 3])

  117. 

  118.     def test_hsvtorgb(self):

  119. 

  120.         self.assertEqual(tl.get_tensor_shape(self.hsvtorgb), [1, 100, 100, 3])

  121. 

  122.     def test_adjustbrightness(self):

  123. 

  124.         self.assertEqual(tl.get_tensor_shape(self.adjustbrightness), [1, 100, 100, 3])

  125. 

  126.     def test_adjustconstrast(self):

  127. 

  128.         self.assertEqual(tl.get_tensor_shape(self.adjustconstrast), [1, 100, 100, 3])

  129. 

  130.     def test_adjusthue(self):

  131. 

  132.         self.assertEqual(tl.get_tensor_shape(self.adjusthue), [1, 100, 100, 3])

  133. 

  134.     def test_adjustsaturation(self):

  135. 

  136.         self.assertEqual(tl.get_tensor_shape(self.adjustsaturation), [1, 100, 100, 3])

  137. 

  138.     def test_crop(self):

  139. 

  140.         self.assertEqual(tl.get_tensor_shape(self.crop), [1, 60, 60, 3])

  141. 

  142.     def test_fliphorizontal(self):

  143. 

  144.         self.assertEqual(tl.get_tensor_shape(self.fliphorizontal), [1, 100, 100, 3])

  145. 

  146.     def test_flipvertical(self):

  147. 

  148.         self.assertEqual(tl.get_tensor_shape(self.flipvertical), [1, 100, 100, 3])

  149. 

  150.     def test_rgbtogray(self):

  151. 

  152.         self.assertEqual(tl.get_tensor_shape(self.rgbtogray), [1, 100, 100, 1])

  153. 

  154.     def test_graytorgb(self):

  155. 

  156.         self.assertEqual(tl.get_tensor_shape(self.graytorgb), [1, 100, 100, 3])

  157. 

  158.     def test_padtoboundingbox(self):

  159. 

  160.         self.assertEqual(tl.get_tensor_shape(self.padtoboundingbox), [1, 150, 150, 3])

  161. 

  162.     def test_pad_1(self):

  163. 

  164.         self.assertEqual(tl.get_tensor_shape(self.pad_1), [1, 120, 120, 3])

  165. 

  166.     def test_pad_2(self):

  167. 

  168.         self.assertEqual(tl.get_tensor_shape(self.pad_2), [1, 120, 120, 3])

  169. 

  170.     def test_pad_3(self):

  171. 

  172.         self.assertEqual(tl.get_tensor_shape(self.pad_3), [1, 130, 170, 3])

  173. 

  174.     def test_standardization_1(self):

  175. 

  176.         self.assertEqual(tl.get_tensor_shape(self.standardization_1), [1, 100, 100, 3])

  177. 

  178.     def test_standardization_2(self):

  179. 

  180.         self.assertEqual(tl.get_tensor_shape(self.standardization_2), [1, 100, 100, 3])

  181. 

  182.     def test_standardization_3(self):

  183. 

  184.         self.assertEqual(tl.get_tensor_shape(self.standardization_3), [1, 100, 100, 3])

  185. 

  186.     def test_randomcontrast(self):

  187. 

  188.         self.assertEqual(tl.get_tensor_shape(self.randomcontrast), [1, 100, 100, 3])

  189. 

  190.     def test_randomhue(self):

  191. 

  192.         self.assertEqual(tl.get_tensor_shape(self.randomhue), [1, 100, 100, 3])

  193. 

  194.     def test_randomsaturation(self):

  195. 

  196.         self.assertEqual(tl.get_tensor_shape(self.randomsaturation), [1, 100, 100, 3])

  197. 

  198.     def test_randomcrop_1(self):

  199. 

  200.         self.assertEqual(tl.get_tensor_shape(self.randomcrop_1), [1, 50, 50, 3])

  201. 

  202.     def test_randomcrop_2(self):

  203. 

  204.         self.assertEqual(tl.get_tensor_shape(self.randomcrop_2), [1, 50, 60, 3])

  205. 

  206.     def test_resize_1(self):

  207. 

  208.         self.assertEqual(tl.get_tensor_shape(self.resize_1), [1, 46, 46, 3])

  209. 

  210.     def test_resize_2(self):

  211. 

  212.         self.assertEqual(tl.get_tensor_shape(self.resize_2), [1, 32, 32, 3])

  213. 

  214.     def test_croporpad(self):

  215. 

  216.         self.assertEqual(tl.get_tensor_shape(self.croporpad), [1, 50, 150, 3])

  217. 

  218.     def test_resizeandpad(self):

  219. 

  220.         self.assertEqual(tl.get_tensor_shape(self.resizeandpad), [1, 50, 150, 3])

  221. 

  222.     def test_rgbtohsv(self):

  223. 

  224.         self.assertEqual(tl.get_tensor_shape(self.rgbtohsv), [1, 100, 100, 3])

  225. 

  226.     def test_transpose(self):

  227. 

  228.         self.assertEqual(tl.get_tensor_shape(self.transpose), [3, 100, 100, 1])

  229. 

  230.     def test_randomrotation(self):

  231. 

  232.         self.assertEqual(tl.get_tensor_shape(self.randomrotation), [100, 100, 3])

  233. 

  234.     def test_randomshift_1(self):

  235. 

  236.         self.assertEqual(tl.get_tensor_shape(self.randomshift_1), [100, 100, 3])

  237. 

  238.     def test_randomshift_2(self):

  239. 

  240.         self.assertEqual(tl.get_tensor_shape(self.randomshift_2), [100, 100, 3])

  241. 

  242.     def test_randoshear(self):

  243. 

  244.         self.assertEqual(tl.get_tensor_shape(self.randomshear), [100, 100, 3])

  245. 

  246.     def test_randomzoom_1(self):

  247. 

  248.         self.assertEqual(tl.get_tensor_shape(self.randomzoom_1), [100, 100, 3])

  249. 

  250.     def test_randomzoom_2(self):

  251. 

  252.         self.assertEqual(tl.get_tensor_shape(self.randomzoom_2), [100, 100, 3])

  253. 

  254.     def test_rescale(self):

  255. 

  256.         self.assertEqual(tl.get_tensor_shape(self.rescale), [1, 100, 100, 3])

  257. 

  258.     def test_randomflipvertical(self):

  259. 

  260.         self.assertEqual(tl.get_tensor_shape(self.randomflipvertical), [1, 100, 100, 3])

  261. 

  262.     def test_randomfliphorizontal(self):

  263. 

  264.         self.assertEqual(tl.get_tensor_shape(self.randomfliphorizontal), [1, 100, 100, 3])

  265. 

  266.     def test_hwc2chw(self):

  267. 

  268.         self.assertEqual(tl.get_tensor_shape(self.hwc2chw), [1, 3, 100, 100])

  269. 

  270.     def test_chw2hwc(self):

  271. 

  272.         self.assertEqual(tl.get_tensor_shape(self.chw2hwc), [1, 100, 100, 3])

  273. 

  274. 

  275. if __name__ == '__main__':

  276. 

  277.     tl.logging.set_verbosity(tl.logging.DEBUG)

  278. 

  279.     unittest.main()