kyxt
/
code_test
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: e9b9500e08
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'e9b9500e08'
${ noResults }
code_test/tests/test_data/test_utils.py

81 lines
2.7 kB
Raw Blame History 

  1. # Copyright (c) OpenMMLab. All rights reserved.

  2. import pytest

  3. 

  4. from mmdet.datasets import get_loading_pipeline, replace_ImageToTensor

  5. 

  6. 

  7. def test_replace_ImageToTensor():

  8.     # with MultiScaleFlipAug

  9.     pipelines = [

  10.         dict(type='LoadImageFromFile'),

  11.         dict(

  12.             type='MultiScaleFlipAug',

  13.             img_scale=(1333, 800),

  14.             flip=False,

  15.             transforms=[

  16.                 dict(type='Resize', keep_ratio=True),

  17.                 dict(type='RandomFlip'),

  18.                 dict(type='Normalize'),

  19.                 dict(type='Pad', size_divisor=32),

  20.                 dict(type='ImageToTensor', keys=['img']),

  21.                 dict(type='Collect', keys=['img']),

  22.             ])

  23.     ]

  24.     expected_pipelines = [

  25.         dict(type='LoadImageFromFile'),

  26.         dict(

  27.             type='MultiScaleFlipAug',

  28.             img_scale=(1333, 800),

  29.             flip=False,

  30.             transforms=[

  31.                 dict(type='Resize', keep_ratio=True),

  32.                 dict(type='RandomFlip'),

  33.                 dict(type='Normalize'),

  34.                 dict(type='Pad', size_divisor=32),

  35.                 dict(type='DefaultFormatBundle'),

  36.                 dict(type='Collect', keys=['img']),

  37.             ])

  38.     ]

  39.     with pytest.warns(UserWarning):

  40.         assert expected_pipelines == replace_ImageToTensor(pipelines)

  41. 

  42.     # without MultiScaleFlipAug

  43.     pipelines = [

  44.         dict(type='LoadImageFromFile'),

  45.         dict(type='Resize', keep_ratio=True),

  46.         dict(type='RandomFlip'),

  47.         dict(type='Normalize'),

  48.         dict(type='Pad', size_divisor=32),

  49.         dict(type='ImageToTensor', keys=['img']),

  50.         dict(type='Collect', keys=['img']),

  51.     ]

  52.     expected_pipelines = [

  53.         dict(type='LoadImageFromFile'),

  54.         dict(type='Resize', keep_ratio=True),

  55.         dict(type='RandomFlip'),

  56.         dict(type='Normalize'),

  57.         dict(type='Pad', size_divisor=32),

  58.         dict(type='DefaultFormatBundle'),

  59.         dict(type='Collect', keys=['img']),

  60.     ]

  61.     with pytest.warns(UserWarning):

  62.         assert expected_pipelines == replace_ImageToTensor(pipelines)

  63. 

  64. 

  65. def test_get_loading_pipeline():

  66.     pipelines = [

  67.         dict(type='LoadImageFromFile'),

  68.         dict(type='LoadAnnotations', with_bbox=True),

  69.         dict(type='Resize', img_scale=(1333, 800), keep_ratio=True),

  70.         dict(type='RandomFlip', flip_ratio=0.5),

  71.         dict(type='Pad', size_divisor=32),

  72.         dict(type='DefaultFormatBundle'),

  73.         dict(type='Collect', keys=['img', 'gt_bboxes', 'gt_labels'])

  74.     ]

  75.     expected_pipelines = [

  76.         dict(type='LoadImageFromFile'),

  77.         dict(type='LoadAnnotations', with_bbox=True)

  78.     ]

  79.     assert expected_pipelines == \

  80.            get_loading_pipeline(pipelines)