File tree 1 file changed +3
-10
lines changed
1 file changed +3
-10
lines changed Original file line number Diff line number Diff line change @@ -105,8 +105,7 @@ def test_dataset_wrapper():
105
105
106
106
img_scale = (60 , 60 )
107
107
pipeline = [
108
- # dict(type='Mosaic', img_scale=img_scale, pad_val=255),
109
- # need to merge mosaic
108
+ dict (type = 'RandomMosaic' , prob = 1 , img_scale = img_scale ),
110
109
dict (type = 'RandomFlip' , prob = 0.5 ),
111
110
dict (type = 'Resize' , img_scale = img_scale , keep_ratio = False ),
112
111
]
@@ -130,14 +129,8 @@ def test_dataset_wrapper():
130
129
classes = classes ,
131
130
palette = palette )
132
131
len_a = 2
133
- cat_ids_list_a = [
134
- np .random .randint (0 , 80 , num ).tolist ()
135
- for num in np .random .randint (1 , 20 , len_a )
136
- ]
137
- dataset_a .data_infos = MagicMock ()
138
- dataset_a .data_infos .__len__ .return_value = len_a
139
- dataset_a .get_cat_ids = MagicMock (
140
- side_effect = lambda idx : cat_ids_list_a [idx ])
132
+ dataset_a .img_infos = MagicMock ()
133
+ dataset_a .img_infos .__len__ .return_value = len_a
141
134
142
135
multi_image_mix_dataset = MultiImageMixDataset (dataset_a , pipeline )
143
136
assert len (multi_image_mix_dataset ) == len (dataset_a )
You can’t perform that action at this time.
0 commit comments