Unlock 30% off on Manual Testing Annual Plans this Holiday Season. 
Use Code: LCMYEARENDEXCL30
Use Code: LCMYEARENDEXCL30Check icon
Copied to Clipboard!
How to use Feature method in ladle
Best JavaScript code snippet using ladle
feature_map_generators_test.py
Source:feature_map_generators_test.py 
1# Copyright 2017 The TensorFlow Authors. All Rights Reserved.2#3# Licensed under the Apache License, Version 2.0 (the "License");4# you may not use this file except in compliance with the License.5# You may obtain a copy of the License at6#7# http://www.apache.org/licenses/LICENSE-2.08#9# Unless required by applicable law or agreed to in writing, software10# distributed under the License is distributed on an "AS IS" BASIS,11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.12# See the License for the specific language governing permissions and13# limitations under the License.14# ==============================================================================15"""Tests for feature map generators."""16from absl.testing import parameterized17import numpy as np18import tensorflow as tf19from google.protobuf import text_format20from object_detection.builders import hyperparams_builder21from object_detection.models import feature_map_generators22from object_detection.protos import hyperparams_pb223INCEPTION_V2_LAYOUT = {24 'from_layer': ['Mixed_3c', 'Mixed_4c', 'Mixed_5c', '', '', ''],25 'layer_depth': [-1, -1, -1, 512, 256, 256],26 'anchor_strides': [16, 32, 64, -1, -1, -1],27 'layer_target_norm': [20.0, -1, -1, -1, -1, -1],28}29INCEPTION_V3_LAYOUT = {30 'from_layer': ['Mixed_5d', 'Mixed_6e', 'Mixed_7c', '', '', ''],31 'layer_depth': [-1, -1, -1, 512, 256, 128],32 'anchor_strides': [16, 32, 64, -1, -1, -1],33 'aspect_ratios': [1.0, 2.0, 1.0/2, 3.0, 1.0/3]34}35EMBEDDED_SSD_MOBILENET_V1_LAYOUT = {36 'from_layer': ['Conv2d_11_pointwise', 'Conv2d_13_pointwise', '', '', ''],37 'layer_depth': [-1, -1, 512, 256, 256],38 'conv_kernel_size': [-1, -1, 3, 3, 2],39}40SSD_MOBILENET_V1_WEIGHT_SHARED_LAYOUT = {41 'from_layer': ['Conv2d_13_pointwise', '', '', ''],42 'layer_depth': [-1, 256, 256, 256],43}44@parameterized.parameters(45 {'use_keras': False},46 {'use_keras': True},47)48class MultiResolutionFeatureMapGeneratorTest(tf.test.TestCase):49 def _build_conv_hyperparams(self):50 conv_hyperparams = hyperparams_pb2.Hyperparams()51 conv_hyperparams_text_proto = """52 regularizer {53 l2_regularizer {54 }55 }56 initializer {57 truncated_normal_initializer {58 }59 }60 """61 text_format.Merge(conv_hyperparams_text_proto, conv_hyperparams)62 return hyperparams_builder.KerasLayerHyperparams(conv_hyperparams)63 def _build_feature_map_generator(self, feature_map_layout, use_keras,64 pool_residual=False):65 if use_keras:66 return feature_map_generators.KerasMultiResolutionFeatureMaps(67 feature_map_layout=feature_map_layout,68 depth_multiplier=1,69 min_depth=32,70 insert_1x1_conv=True,71 freeze_batchnorm=False,72 is_training=True,73 conv_hyperparams=self._build_conv_hyperparams(),74 name='FeatureMaps'75 )76 else:77 def feature_map_generator(image_features):78 return feature_map_generators.multi_resolution_feature_maps(79 feature_map_layout=feature_map_layout,80 depth_multiplier=1,81 min_depth=32,82 insert_1x1_conv=True,83 image_features=image_features,84 pool_residual=pool_residual)85 return feature_map_generator86 def test_get_expected_feature_map_shapes_with_inception_v2(self, use_keras):87 image_features = {88 'Mixed_3c': tf.random_uniform([4, 28, 28, 256], dtype=tf.float32),89 'Mixed_4c': tf.random_uniform([4, 14, 14, 576], dtype=tf.float32),90 'Mixed_5c': tf.random_uniform([4, 7, 7, 1024], dtype=tf.float32)91 }92 feature_map_generator = self._build_feature_map_generator(93 feature_map_layout=INCEPTION_V2_LAYOUT,94 use_keras=use_keras95 )96 feature_maps = feature_map_generator(image_features)97 expected_feature_map_shapes = {98 'Mixed_3c': (4, 28, 28, 256),99 'Mixed_4c': (4, 14, 14, 576),100 'Mixed_5c': (4, 7, 7, 1024),101 'Mixed_5c_2_Conv2d_3_3x3_s2_512': (4, 4, 4, 512),102 'Mixed_5c_2_Conv2d_4_3x3_s2_256': (4, 2, 2, 256),103 'Mixed_5c_2_Conv2d_5_3x3_s2_256': (4, 1, 1, 256)}104 init_op = tf.global_variables_initializer()105 with self.test_session() as sess:106 sess.run(init_op)107 out_feature_maps = sess.run(feature_maps)108 out_feature_map_shapes = dict(109 (key, value.shape) for key, value in out_feature_maps.items())110 self.assertDictEqual(expected_feature_map_shapes, out_feature_map_shapes)111 def test_get_expected_feature_map_shapes_with_inception_v2_use_depthwise(112 self, use_keras):113 image_features = {114 'Mixed_3c': tf.random_uniform([4, 28, 28, 256], dtype=tf.float32),115 'Mixed_4c': tf.random_uniform([4, 14, 14, 576], dtype=tf.float32),116 'Mixed_5c': tf.random_uniform([4, 7, 7, 1024], dtype=tf.float32)117 }118 layout_copy = INCEPTION_V2_LAYOUT.copy()119 layout_copy['use_depthwise'] = True120 feature_map_generator = self._build_feature_map_generator(121 feature_map_layout=layout_copy,122 use_keras=use_keras123 )124 feature_maps = feature_map_generator(image_features)125 expected_feature_map_shapes = {126 'Mixed_3c': (4, 28, 28, 256),127 'Mixed_4c': (4, 14, 14, 576),128 'Mixed_5c': (4, 7, 7, 1024),129 'Mixed_5c_2_Conv2d_3_3x3_s2_512': (4, 4, 4, 512),130 'Mixed_5c_2_Conv2d_4_3x3_s2_256': (4, 2, 2, 256),131 'Mixed_5c_2_Conv2d_5_3x3_s2_256': (4, 1, 1, 256)}132 init_op = tf.global_variables_initializer()133 with self.test_session() as sess:134 sess.run(init_op)135 out_feature_maps = sess.run(feature_maps)136 out_feature_map_shapes = dict(137 (key, value.shape) for key, value in out_feature_maps.items())138 self.assertDictEqual(expected_feature_map_shapes, out_feature_map_shapes)139 def test_get_expected_feature_map_shapes_use_explicit_padding(140 self, use_keras):141 image_features = {142 'Mixed_3c': tf.random_uniform([4, 28, 28, 256], dtype=tf.float32),143 'Mixed_4c': tf.random_uniform([4, 14, 14, 576], dtype=tf.float32),144 'Mixed_5c': tf.random_uniform([4, 7, 7, 1024], dtype=tf.float32)145 }146 layout_copy = INCEPTION_V2_LAYOUT.copy()147 layout_copy['use_explicit_padding'] = True148 feature_map_generator = self._build_feature_map_generator(149 feature_map_layout=layout_copy,150 use_keras=use_keras151 )152 feature_maps = feature_map_generator(image_features)153 expected_feature_map_shapes = {154 'Mixed_3c': (4, 28, 28, 256),155 'Mixed_4c': (4, 14, 14, 576),156 'Mixed_5c': (4, 7, 7, 1024),157 'Mixed_5c_2_Conv2d_3_3x3_s2_512': (4, 4, 4, 512),158 'Mixed_5c_2_Conv2d_4_3x3_s2_256': (4, 2, 2, 256),159 'Mixed_5c_2_Conv2d_5_3x3_s2_256': (4, 1, 1, 256)}160 init_op = tf.global_variables_initializer()161 with self.test_session() as sess:162 sess.run(init_op)163 out_feature_maps = sess.run(feature_maps)164 out_feature_map_shapes = dict(165 (key, value.shape) for key, value in out_feature_maps.items())166 self.assertDictEqual(expected_feature_map_shapes, out_feature_map_shapes)167 def test_get_expected_feature_map_shapes_with_inception_v3(self, use_keras):168 image_features = {169 'Mixed_5d': tf.random_uniform([4, 35, 35, 256], dtype=tf.float32),170 'Mixed_6e': tf.random_uniform([4, 17, 17, 576], dtype=tf.float32),171 'Mixed_7c': tf.random_uniform([4, 8, 8, 1024], dtype=tf.float32)172 }173 feature_map_generator = self._build_feature_map_generator(174 feature_map_layout=INCEPTION_V3_LAYOUT,175 use_keras=use_keras176 )177 feature_maps = feature_map_generator(image_features)178 expected_feature_map_shapes = {179 'Mixed_5d': (4, 35, 35, 256),180 'Mixed_6e': (4, 17, 17, 576),181 'Mixed_7c': (4, 8, 8, 1024),182 'Mixed_7c_2_Conv2d_3_3x3_s2_512': (4, 4, 4, 512),183 'Mixed_7c_2_Conv2d_4_3x3_s2_256': (4, 2, 2, 256),184 'Mixed_7c_2_Conv2d_5_3x3_s2_128': (4, 1, 1, 128)}185 init_op = tf.global_variables_initializer()186 with self.test_session() as sess:187 sess.run(init_op)188 out_feature_maps = sess.run(feature_maps)189 out_feature_map_shapes = dict(190 (key, value.shape) for key, value in out_feature_maps.items())191 self.assertDictEqual(expected_feature_map_shapes, out_feature_map_shapes)192 def test_get_expected_feature_map_shapes_with_embedded_ssd_mobilenet_v1(193 self, use_keras):194 image_features = {195 'Conv2d_11_pointwise': tf.random_uniform([4, 16, 16, 512],196 dtype=tf.float32),197 'Conv2d_13_pointwise': tf.random_uniform([4, 8, 8, 1024],198 dtype=tf.float32),199 }200 feature_map_generator = self._build_feature_map_generator(201 feature_map_layout=EMBEDDED_SSD_MOBILENET_V1_LAYOUT,202 use_keras=use_keras203 )204 feature_maps = feature_map_generator(image_features)205 expected_feature_map_shapes = {206 'Conv2d_11_pointwise': (4, 16, 16, 512),207 'Conv2d_13_pointwise': (4, 8, 8, 1024),208 'Conv2d_13_pointwise_2_Conv2d_2_3x3_s2_512': (4, 4, 4, 512),209 'Conv2d_13_pointwise_2_Conv2d_3_3x3_s2_256': (4, 2, 2, 256),210 'Conv2d_13_pointwise_2_Conv2d_4_2x2_s2_256': (4, 1, 1, 256)}211 init_op = tf.global_variables_initializer()212 with self.test_session() as sess:213 sess.run(init_op)214 out_feature_maps = sess.run(feature_maps)215 out_feature_map_shapes = dict(216 (key, value.shape) for key, value in out_feature_maps.items())217 self.assertDictEqual(expected_feature_map_shapes, out_feature_map_shapes)218 def test_feature_map_shapes_with_pool_residual_ssd_mobilenet_v1(219 self, use_keras):220 image_features = {221 'Conv2d_13_pointwise': tf.random_uniform([4, 8, 8, 1024],222 dtype=tf.float32),223 }224 feature_map_generator = self._build_feature_map_generator(225 feature_map_layout=SSD_MOBILENET_V1_WEIGHT_SHARED_LAYOUT,226 use_keras=use_keras,227 pool_residual=True228 )229 feature_maps = feature_map_generator(image_features)230 expected_feature_map_shapes = {231 'Conv2d_13_pointwise': (4, 8, 8, 1024),232 'Conv2d_13_pointwise_2_Conv2d_1_3x3_s2_256': (4, 4, 4, 256),233 'Conv2d_13_pointwise_2_Conv2d_2_3x3_s2_256': (4, 2, 2, 256),234 'Conv2d_13_pointwise_2_Conv2d_3_3x3_s2_256': (4, 1, 1, 256)}235 init_op = tf.global_variables_initializer()236 with self.test_session() as sess:237 sess.run(init_op)238 out_feature_maps = sess.run(feature_maps)239 out_feature_map_shapes = dict(240 (key, value.shape) for key, value in out_feature_maps.items())241 self.assertDictEqual(expected_feature_map_shapes, out_feature_map_shapes)242 def test_get_expected_variable_names_with_inception_v2(self, use_keras):243 image_features = {244 'Mixed_3c': tf.random_uniform([4, 28, 28, 256], dtype=tf.float32),245 'Mixed_4c': tf.random_uniform([4, 14, 14, 576], dtype=tf.float32),246 'Mixed_5c': tf.random_uniform([4, 7, 7, 1024], dtype=tf.float32)247 }248 feature_map_generator = self._build_feature_map_generator(249 feature_map_layout=INCEPTION_V2_LAYOUT,250 use_keras=use_keras251 )252 feature_maps = feature_map_generator(image_features)253 expected_slim_variables = set([254 'Mixed_5c_1_Conv2d_3_1x1_256/weights',255 'Mixed_5c_1_Conv2d_3_1x1_256/biases',256 'Mixed_5c_2_Conv2d_3_3x3_s2_512/weights',257 'Mixed_5c_2_Conv2d_3_3x3_s2_512/biases',258 'Mixed_5c_1_Conv2d_4_1x1_128/weights',259 'Mixed_5c_1_Conv2d_4_1x1_128/biases',260 'Mixed_5c_2_Conv2d_4_3x3_s2_256/weights',261 'Mixed_5c_2_Conv2d_4_3x3_s2_256/biases',262 'Mixed_5c_1_Conv2d_5_1x1_128/weights',263 'Mixed_5c_1_Conv2d_5_1x1_128/biases',264 'Mixed_5c_2_Conv2d_5_3x3_s2_256/weights',265 'Mixed_5c_2_Conv2d_5_3x3_s2_256/biases',266 ])267 expected_keras_variables = set([268 'FeatureMaps/Mixed_5c_1_Conv2d_3_1x1_256_conv/kernel',269 'FeatureMaps/Mixed_5c_1_Conv2d_3_1x1_256_conv/bias',270 'FeatureMaps/Mixed_5c_2_Conv2d_3_3x3_s2_512_conv/kernel',271 'FeatureMaps/Mixed_5c_2_Conv2d_3_3x3_s2_512_conv/bias',272 'FeatureMaps/Mixed_5c_1_Conv2d_4_1x1_128_conv/kernel',273 'FeatureMaps/Mixed_5c_1_Conv2d_4_1x1_128_conv/bias',274 'FeatureMaps/Mixed_5c_2_Conv2d_4_3x3_s2_256_conv/kernel',275 'FeatureMaps/Mixed_5c_2_Conv2d_4_3x3_s2_256_conv/bias',276 'FeatureMaps/Mixed_5c_1_Conv2d_5_1x1_128_conv/kernel',277 'FeatureMaps/Mixed_5c_1_Conv2d_5_1x1_128_conv/bias',278 'FeatureMaps/Mixed_5c_2_Conv2d_5_3x3_s2_256_conv/kernel',279 'FeatureMaps/Mixed_5c_2_Conv2d_5_3x3_s2_256_conv/bias',280 ])281 init_op = tf.global_variables_initializer()282 with self.test_session() as sess:283 sess.run(init_op)284 sess.run(feature_maps)285 actual_variable_set = set(286 [var.op.name for var in tf.trainable_variables()])287 if use_keras:288 self.assertSetEqual(expected_keras_variables, actual_variable_set)289 else:290 self.assertSetEqual(expected_slim_variables, actual_variable_set)291 def test_get_expected_variable_names_with_inception_v2_use_depthwise(292 self,293 use_keras):294 image_features = {295 'Mixed_3c': tf.random_uniform([4, 28, 28, 256], dtype=tf.float32),296 'Mixed_4c': tf.random_uniform([4, 14, 14, 576], dtype=tf.float32),297 'Mixed_5c': tf.random_uniform([4, 7, 7, 1024], dtype=tf.float32)298 }299 layout_copy = INCEPTION_V2_LAYOUT.copy()300 layout_copy['use_depthwise'] = True301 feature_map_generator = self._build_feature_map_generator(302 feature_map_layout=layout_copy,303 use_keras=use_keras304 )305 feature_maps = feature_map_generator(image_features)306 expected_slim_variables = set([307 'Mixed_5c_1_Conv2d_3_1x1_256/weights',308 'Mixed_5c_1_Conv2d_3_1x1_256/biases',309 'Mixed_5c_2_Conv2d_3_3x3_s2_512_depthwise/depthwise_weights',310 'Mixed_5c_2_Conv2d_3_3x3_s2_512_depthwise/biases',311 'Mixed_5c_2_Conv2d_3_3x3_s2_512/weights',312 'Mixed_5c_2_Conv2d_3_3x3_s2_512/biases',313 'Mixed_5c_1_Conv2d_4_1x1_128/weights',314 'Mixed_5c_1_Conv2d_4_1x1_128/biases',315 'Mixed_5c_2_Conv2d_4_3x3_s2_256_depthwise/depthwise_weights',316 'Mixed_5c_2_Conv2d_4_3x3_s2_256_depthwise/biases',317 'Mixed_5c_2_Conv2d_4_3x3_s2_256/weights',318 'Mixed_5c_2_Conv2d_4_3x3_s2_256/biases',319 'Mixed_5c_1_Conv2d_5_1x1_128/weights',320 'Mixed_5c_1_Conv2d_5_1x1_128/biases',321 'Mixed_5c_2_Conv2d_5_3x3_s2_256_depthwise/depthwise_weights',322 'Mixed_5c_2_Conv2d_5_3x3_s2_256_depthwise/biases',323 'Mixed_5c_2_Conv2d_5_3x3_s2_256/weights',324 'Mixed_5c_2_Conv2d_5_3x3_s2_256/biases',325 ])326 expected_keras_variables = set([327 'FeatureMaps/Mixed_5c_1_Conv2d_3_1x1_256_conv/kernel',328 'FeatureMaps/Mixed_5c_1_Conv2d_3_1x1_256_conv/bias',329 ('FeatureMaps/Mixed_5c_2_Conv2d_3_3x3_s2_512_depthwise_conv/'330 'depthwise_kernel'),331 ('FeatureMaps/Mixed_5c_2_Conv2d_3_3x3_s2_512_depthwise_conv/'332 'bias'),333 'FeatureMaps/Mixed_5c_2_Conv2d_3_3x3_s2_512_conv/kernel',334 'FeatureMaps/Mixed_5c_2_Conv2d_3_3x3_s2_512_conv/bias',335 'FeatureMaps/Mixed_5c_1_Conv2d_4_1x1_128_conv/kernel',336 'FeatureMaps/Mixed_5c_1_Conv2d_4_1x1_128_conv/bias',337 ('FeatureMaps/Mixed_5c_2_Conv2d_4_3x3_s2_256_depthwise_conv/'338 'depthwise_kernel'),339 ('FeatureMaps/Mixed_5c_2_Conv2d_4_3x3_s2_256_depthwise_conv/'340 'bias'),341 'FeatureMaps/Mixed_5c_2_Conv2d_4_3x3_s2_256_conv/kernel',342 'FeatureMaps/Mixed_5c_2_Conv2d_4_3x3_s2_256_conv/bias',343 'FeatureMaps/Mixed_5c_1_Conv2d_5_1x1_128_conv/kernel',344 'FeatureMaps/Mixed_5c_1_Conv2d_5_1x1_128_conv/bias',345 ('FeatureMaps/Mixed_5c_2_Conv2d_5_3x3_s2_256_depthwise_conv/'346 'depthwise_kernel'),347 ('FeatureMaps/Mixed_5c_2_Conv2d_5_3x3_s2_256_depthwise_conv/'348 'bias'),349 'FeatureMaps/Mixed_5c_2_Conv2d_5_3x3_s2_256_conv/kernel',350 'FeatureMaps/Mixed_5c_2_Conv2d_5_3x3_s2_256_conv/bias',351 ])352 init_op = tf.global_variables_initializer()353 with self.test_session() as sess:354 sess.run(init_op)355 sess.run(feature_maps)356 actual_variable_set = set(357 [var.op.name for var in tf.trainable_variables()])358 if use_keras:359 self.assertSetEqual(expected_keras_variables, actual_variable_set)360 else:361 self.assertSetEqual(expected_slim_variables, actual_variable_set)362@parameterized.parameters({'use_native_resize_op': True, 'use_keras': False},363 {'use_native_resize_op': False, 'use_keras': False},364 {'use_native_resize_op': True, 'use_keras': True},365 {'use_native_resize_op': False, 'use_keras': True})366class FPNFeatureMapGeneratorTest(tf.test.TestCase, parameterized.TestCase):367 def _build_conv_hyperparams(self):368 conv_hyperparams = hyperparams_pb2.Hyperparams()369 conv_hyperparams_text_proto = """370 regularizer {371 l2_regularizer {372 }373 }374 initializer {375 truncated_normal_initializer {376 }377 }378 """379 text_format.Merge(conv_hyperparams_text_proto, conv_hyperparams)380 return hyperparams_builder.KerasLayerHyperparams(conv_hyperparams)381 def _build_feature_map_generator(382 self, image_features, depth, use_keras, use_bounded_activations=False,383 use_native_resize_op=False, use_explicit_padding=False,384 use_depthwise=False):385 if use_keras:386 return feature_map_generators.KerasFpnTopDownFeatureMaps(387 num_levels=len(image_features),388 depth=depth,389 is_training=True,390 conv_hyperparams=self._build_conv_hyperparams(),391 freeze_batchnorm=False,392 use_depthwise=use_depthwise,393 use_explicit_padding=use_explicit_padding,394 use_bounded_activations=use_bounded_activations,395 use_native_resize_op=use_native_resize_op,396 scope=None,397 name='FeatureMaps',398 )399 else:400 def feature_map_generator(image_features):401 return feature_map_generators.fpn_top_down_feature_maps(402 image_features=image_features,403 depth=depth,404 use_depthwise=use_depthwise,405 use_explicit_padding=use_explicit_padding,406 use_bounded_activations=use_bounded_activations,407 use_native_resize_op=use_native_resize_op)408 return feature_map_generator409 def test_get_expected_feature_map_shapes(410 self, use_native_resize_op, use_keras):411 image_features = [412 ('block2', tf.random_uniform([4, 8, 8, 256], dtype=tf.float32)),413 ('block3', tf.random_uniform([4, 4, 4, 256], dtype=tf.float32)),414 ('block4', tf.random_uniform([4, 2, 2, 256], dtype=tf.float32)),415 ('block5', tf.random_uniform([4, 1, 1, 256], dtype=tf.float32))416 ]417 feature_map_generator = self._build_feature_map_generator(418 image_features=image_features,419 depth=128,420 use_keras=use_keras,421 use_native_resize_op=use_native_resize_op)422 feature_maps = feature_map_generator(image_features)423 expected_feature_map_shapes = {424 'top_down_block2': (4, 8, 8, 128),425 'top_down_block3': (4, 4, 4, 128),426 'top_down_block4': (4, 2, 2, 128),427 'top_down_block5': (4, 1, 1, 128)428 }429 init_op = tf.global_variables_initializer()430 with self.test_session() as sess:431 sess.run(init_op)432 out_feature_maps = sess.run(feature_maps)433 out_feature_map_shapes = {key: value.shape434 for key, value in out_feature_maps.items()}435 self.assertDictEqual(out_feature_map_shapes, expected_feature_map_shapes)436 def test_get_expected_feature_map_shapes_with_explicit_padding(437 self, use_native_resize_op, use_keras):438 image_features = [439 ('block2', tf.random_uniform([4, 8, 8, 256], dtype=tf.float32)),440 ('block3', tf.random_uniform([4, 4, 4, 256], dtype=tf.float32)),441 ('block4', tf.random_uniform([4, 2, 2, 256], dtype=tf.float32)),442 ('block5', tf.random_uniform([4, 1, 1, 256], dtype=tf.float32))443 ]444 feature_map_generator = self._build_feature_map_generator(445 image_features=image_features,446 depth=128,447 use_keras=use_keras,448 use_explicit_padding=True,449 use_native_resize_op=use_native_resize_op)450 feature_maps = feature_map_generator(image_features)451 expected_feature_map_shapes = {452 'top_down_block2': (4, 8, 8, 128),453 'top_down_block3': (4, 4, 4, 128),454 'top_down_block4': (4, 2, 2, 128),455 'top_down_block5': (4, 1, 1, 128)456 }457 init_op = tf.global_variables_initializer()458 with self.test_session() as sess:459 sess.run(init_op)460 out_feature_maps = sess.run(feature_maps)461 out_feature_map_shapes = {key: value.shape462 for key, value in out_feature_maps.items()}463 self.assertDictEqual(out_feature_map_shapes, expected_feature_map_shapes)464 def test_use_bounded_activations_add_operations(465 self, use_native_resize_op, use_keras):466 tf_graph = tf.Graph()467 with tf_graph.as_default():468 image_features = [('block2',469 tf.random_uniform([4, 8, 8, 256], dtype=tf.float32)),470 ('block3',471 tf.random_uniform([4, 4, 4, 256], dtype=tf.float32)),472 ('block4',473 tf.random_uniform([4, 2, 2, 256], dtype=tf.float32)),474 ('block5',475 tf.random_uniform([4, 1, 1, 256], dtype=tf.float32))]476 feature_map_generator = self._build_feature_map_generator(477 image_features=image_features,478 depth=128,479 use_keras=use_keras,480 use_bounded_activations=True,481 use_native_resize_op=use_native_resize_op)482 feature_map_generator(image_features)483 if use_keras:484 expected_added_operations = dict.fromkeys([485 'FeatureMaps/top_down/clip_by_value/clip_by_value',486 'FeatureMaps/top_down/clip_by_value_1/clip_by_value',487 'FeatureMaps/top_down/clip_by_value_2/clip_by_value',488 'FeatureMaps/top_down/clip_by_value_3/clip_by_value',489 'FeatureMaps/top_down/clip_by_value_4/clip_by_value',490 'FeatureMaps/top_down/clip_by_value_5/clip_by_value',491 'FeatureMaps/top_down/clip_by_value_6/clip_by_value',492 ])493 else:494 expected_added_operations = dict.fromkeys([495 'top_down/clip_by_value', 'top_down/clip_by_value_1',496 'top_down/clip_by_value_2', 'top_down/clip_by_value_3',497 'top_down/clip_by_value_4', 'top_down/clip_by_value_5',498 'top_down/clip_by_value_6'499 ])500 op_names = {op.name: None for op in tf_graph.get_operations()}501 self.assertDictContainsSubset(expected_added_operations, op_names)502 def test_use_bounded_activations_clip_value(503 self, use_native_resize_op, use_keras):504 tf_graph = tf.Graph()505 with tf_graph.as_default():506 image_features = [507 ('block2', 255 * tf.ones([4, 8, 8, 256], dtype=tf.float32)),508 ('block3', 255 * tf.ones([4, 4, 4, 256], dtype=tf.float32)),509 ('block4', 255 * tf.ones([4, 2, 2, 256], dtype=tf.float32)),510 ('block5', 255 * tf.ones([4, 1, 1, 256], dtype=tf.float32))511 ]512 feature_map_generator = self._build_feature_map_generator(513 image_features=image_features,514 depth=128,515 use_keras=use_keras,516 use_bounded_activations=True,517 use_native_resize_op=use_native_resize_op)518 feature_map_generator(image_features)519 if use_keras:520 expected_clip_by_value_ops = dict.fromkeys([521 'FeatureMaps/top_down/clip_by_value/clip_by_value',522 'FeatureMaps/top_down/clip_by_value_1/clip_by_value',523 'FeatureMaps/top_down/clip_by_value_2/clip_by_value',524 'FeatureMaps/top_down/clip_by_value_3/clip_by_value',525 'FeatureMaps/top_down/clip_by_value_4/clip_by_value',526 'FeatureMaps/top_down/clip_by_value_5/clip_by_value',527 'FeatureMaps/top_down/clip_by_value_6/clip_by_value',528 ])529 else:530 expected_clip_by_value_ops = [531 'top_down/clip_by_value', 'top_down/clip_by_value_1',532 'top_down/clip_by_value_2', 'top_down/clip_by_value_3',533 'top_down/clip_by_value_4', 'top_down/clip_by_value_5',534 'top_down/clip_by_value_6'535 ]536 # Gathers activation tensors before and after clip_by_value operations.537 activations = {}538 for clip_by_value_op in expected_clip_by_value_ops:539 clip_input_tensor = tf_graph.get_operation_by_name(540 '{}/Minimum'.format(clip_by_value_op)).inputs[0]541 clip_output_tensor = tf_graph.get_tensor_by_name(542 '{}:0'.format(clip_by_value_op))543 activations.update({544 'before_{}'.format(clip_by_value_op): clip_input_tensor,545 'after_{}'.format(clip_by_value_op): clip_output_tensor,546 })547 expected_lower_bound = -feature_map_generators.ACTIVATION_BOUND548 expected_upper_bound = feature_map_generators.ACTIVATION_BOUND549 init_op = tf.global_variables_initializer()550 with self.test_session() as session:551 session.run(init_op)552 activations_output = session.run(activations)553 for clip_by_value_op in expected_clip_by_value_ops:554 # Before clipping, activations are beyound the expected bound because555 # of large input image_features values.556 activations_before_clipping = (557 activations_output['before_{}'.format(clip_by_value_op)])558 before_clipping_lower_bound = np.amin(activations_before_clipping)559 before_clipping_upper_bound = np.amax(activations_before_clipping)560 self.assertLessEqual(before_clipping_lower_bound,561 expected_lower_bound)562 self.assertGreaterEqual(before_clipping_upper_bound,563 expected_upper_bound)564 # After clipping, activations are bounded as expectation.565 activations_after_clipping = (566 activations_output['after_{}'.format(clip_by_value_op)])567 after_clipping_lower_bound = np.amin(activations_after_clipping)568 after_clipping_upper_bound = np.amax(activations_after_clipping)569 self.assertGreaterEqual(after_clipping_lower_bound,570 expected_lower_bound)571 self.assertLessEqual(after_clipping_upper_bound, expected_upper_bound)572 def test_get_expected_feature_map_shapes_with_depthwise(573 self, use_native_resize_op, use_keras):574 image_features = [575 ('block2', tf.random_uniform([4, 8, 8, 256], dtype=tf.float32)),576 ('block3', tf.random_uniform([4, 4, 4, 256], dtype=tf.float32)),577 ('block4', tf.random_uniform([4, 2, 2, 256], dtype=tf.float32)),578 ('block5', tf.random_uniform([4, 1, 1, 256], dtype=tf.float32))579 ]580 feature_map_generator = self._build_feature_map_generator(581 image_features=image_features,582 depth=128,583 use_keras=use_keras,584 use_depthwise=True,585 use_native_resize_op=use_native_resize_op)586 feature_maps = feature_map_generator(image_features)587 expected_feature_map_shapes = {588 'top_down_block2': (4, 8, 8, 128),589 'top_down_block3': (4, 4, 4, 128),590 'top_down_block4': (4, 2, 2, 128),591 'top_down_block5': (4, 1, 1, 128)592 }593 init_op = tf.global_variables_initializer()594 with self.test_session() as sess:595 sess.run(init_op)596 out_feature_maps = sess.run(feature_maps)597 out_feature_map_shapes = {key: value.shape598 for key, value in out_feature_maps.items()}599 self.assertDictEqual(out_feature_map_shapes, expected_feature_map_shapes)600 def test_get_expected_variable_names(601 self, use_native_resize_op, use_keras):602 image_features = [603 ('block2', tf.random_uniform([4, 8, 8, 256], dtype=tf.float32)),604 ('block3', tf.random_uniform([4, 4, 4, 256], dtype=tf.float32)),605 ('block4', tf.random_uniform([4, 2, 2, 256], dtype=tf.float32)),606 ('block5', tf.random_uniform([4, 1, 1, 256], dtype=tf.float32))607 ]608 feature_map_generator = self._build_feature_map_generator(609 image_features=image_features,610 depth=128,611 use_keras=use_keras,612 use_native_resize_op=use_native_resize_op)613 feature_maps = feature_map_generator(image_features)614 expected_slim_variables = set([615 'projection_1/weights',616 'projection_1/biases',617 'projection_2/weights',618 'projection_2/biases',619 'projection_3/weights',620 'projection_3/biases',621 'projection_4/weights',622 'projection_4/biases',623 'smoothing_1/weights',624 'smoothing_1/biases',625 'smoothing_2/weights',626 'smoothing_2/biases',627 'smoothing_3/weights',628 'smoothing_3/biases',629 ])630 expected_keras_variables = set([631 'FeatureMaps/top_down/projection_1/kernel',632 'FeatureMaps/top_down/projection_1/bias',633 'FeatureMaps/top_down/projection_2/kernel',634 'FeatureMaps/top_down/projection_2/bias',635 'FeatureMaps/top_down/projection_3/kernel',636 'FeatureMaps/top_down/projection_3/bias',637 'FeatureMaps/top_down/projection_4/kernel',638 'FeatureMaps/top_down/projection_4/bias',639 'FeatureMaps/top_down/smoothing_1_conv/kernel',640 'FeatureMaps/top_down/smoothing_1_conv/bias',641 'FeatureMaps/top_down/smoothing_2_conv/kernel',642 'FeatureMaps/top_down/smoothing_2_conv/bias',643 'FeatureMaps/top_down/smoothing_3_conv/kernel',644 'FeatureMaps/top_down/smoothing_3_conv/bias'645 ])646 init_op = tf.global_variables_initializer()647 with self.test_session() as sess:648 sess.run(init_op)649 sess.run(feature_maps)650 actual_variable_set = set(651 [var.op.name for var in tf.trainable_variables()])652 if use_keras:653 self.assertSetEqual(expected_keras_variables, actual_variable_set)654 else:655 self.assertSetEqual(expected_slim_variables, actual_variable_set)656 def test_get_expected_variable_names_with_depthwise(657 self, use_native_resize_op, use_keras):658 image_features = [659 ('block2', tf.random_uniform([4, 8, 8, 256], dtype=tf.float32)),660 ('block3', tf.random_uniform([4, 4, 4, 256], dtype=tf.float32)),661 ('block4', tf.random_uniform([4, 2, 2, 256], dtype=tf.float32)),662 ('block5', tf.random_uniform([4, 1, 1, 256], dtype=tf.float32))663 ]664 feature_map_generator = self._build_feature_map_generator(665 image_features=image_features,666 depth=128,667 use_keras=use_keras,668 use_depthwise=True,669 use_native_resize_op=use_native_resize_op)670 feature_maps = feature_map_generator(image_features)671 expected_slim_variables = set([672 'projection_1/weights',673 'projection_1/biases',674 'projection_2/weights',675 'projection_2/biases',676 'projection_3/weights',677 'projection_3/biases',678 'projection_4/weights',679 'projection_4/biases',680 'smoothing_1/depthwise_weights',681 'smoothing_1/pointwise_weights',682 'smoothing_1/biases',683 'smoothing_2/depthwise_weights',684 'smoothing_2/pointwise_weights',685 'smoothing_2/biases',686 'smoothing_3/depthwise_weights',687 'smoothing_3/pointwise_weights',688 'smoothing_3/biases',689 ])690 expected_keras_variables = set([691 'FeatureMaps/top_down/projection_1/kernel',692 'FeatureMaps/top_down/projection_1/bias',693 'FeatureMaps/top_down/projection_2/kernel',694 'FeatureMaps/top_down/projection_2/bias',695 'FeatureMaps/top_down/projection_3/kernel',696 'FeatureMaps/top_down/projection_3/bias',697 'FeatureMaps/top_down/projection_4/kernel',698 'FeatureMaps/top_down/projection_4/bias',699 'FeatureMaps/top_down/smoothing_1_depthwise_conv/depthwise_kernel',700 'FeatureMaps/top_down/smoothing_1_depthwise_conv/pointwise_kernel',701 'FeatureMaps/top_down/smoothing_1_depthwise_conv/bias',702 'FeatureMaps/top_down/smoothing_2_depthwise_conv/depthwise_kernel',703 'FeatureMaps/top_down/smoothing_2_depthwise_conv/pointwise_kernel',704 'FeatureMaps/top_down/smoothing_2_depthwise_conv/bias',705 'FeatureMaps/top_down/smoothing_3_depthwise_conv/depthwise_kernel',706 'FeatureMaps/top_down/smoothing_3_depthwise_conv/pointwise_kernel',707 'FeatureMaps/top_down/smoothing_3_depthwise_conv/bias'708 ])709 init_op = tf.global_variables_initializer()710 with self.test_session() as sess:711 sess.run(init_op)712 sess.run(feature_maps)713 actual_variable_set = set(714 [var.op.name for var in tf.trainable_variables()])715 if use_keras:716 self.assertSetEqual(expected_keras_variables, actual_variable_set)717 else:718 self.assertSetEqual(expected_slim_variables, actual_variable_set)719class GetDepthFunctionTest(tf.test.TestCase):720 def test_return_min_depth_when_multiplier_is_small(self):721 depth_fn = feature_map_generators.get_depth_fn(depth_multiplier=0.5,722 min_depth=16)723 self.assertEqual(depth_fn(16), 16)724 def test_return_correct_depth_with_multiplier(self):725 depth_fn = feature_map_generators.get_depth_fn(depth_multiplier=0.5,726 min_depth=16)727 self.assertEqual(depth_fn(64), 32)728@parameterized.parameters(729 {'replace_pool_with_conv': False},730 {'replace_pool_with_conv': True},731)732class PoolingPyramidFeatureMapGeneratorTest(tf.test.TestCase):733 def test_get_expected_feature_map_shapes(self, replace_pool_with_conv):734 image_features = {735 'image_features': tf.random_uniform([4, 19, 19, 1024])736 }737 feature_maps = feature_map_generators.pooling_pyramid_feature_maps(738 base_feature_map_depth=1024,739 num_layers=6,740 image_features=image_features,741 replace_pool_with_conv=replace_pool_with_conv)742 expected_pool_feature_map_shapes = {743 'Base_Conv2d_1x1_1024': (4, 19, 19, 1024),744 'MaxPool2d_0_2x2': (4, 10, 10, 1024),745 'MaxPool2d_1_2x2': (4, 5, 5, 1024),746 'MaxPool2d_2_2x2': (4, 3, 3, 1024),747 'MaxPool2d_3_2x2': (4, 2, 2, 1024),748 'MaxPool2d_4_2x2': (4, 1, 1, 1024),749 }750 expected_conv_feature_map_shapes = {751 'Base_Conv2d_1x1_1024': (4, 19, 19, 1024),752 'Conv2d_0_3x3_s2_1024': (4, 10, 10, 1024),753 'Conv2d_1_3x3_s2_1024': (4, 5, 5, 1024),754 'Conv2d_2_3x3_s2_1024': (4, 3, 3, 1024),755 'Conv2d_3_3x3_s2_1024': (4, 2, 2, 1024),756 'Conv2d_4_3x3_s2_1024': (4, 1, 1, 1024),757 }758 init_op = tf.global_variables_initializer()759 with self.test_session() as sess:760 sess.run(init_op)761 out_feature_maps = sess.run(feature_maps)762 out_feature_map_shapes = {key: value.shape763 for key, value in out_feature_maps.items()}764 if replace_pool_with_conv:765 self.assertDictEqual(expected_conv_feature_map_shapes,766 out_feature_map_shapes)767 else:768 self.assertDictEqual(expected_pool_feature_map_shapes,769 out_feature_map_shapes)770 def test_get_expected_variable_names(self, replace_pool_with_conv):771 image_features = {772 'image_features': tf.random_uniform([4, 19, 19, 1024])773 }774 feature_maps = feature_map_generators.pooling_pyramid_feature_maps(775 base_feature_map_depth=1024,776 num_layers=6,777 image_features=image_features,778 replace_pool_with_conv=replace_pool_with_conv)779 expected_pool_variables = set([780 'Base_Conv2d_1x1_1024/weights',781 'Base_Conv2d_1x1_1024/biases',782 ])783 expected_conv_variables = set([784 'Base_Conv2d_1x1_1024/weights',785 'Base_Conv2d_1x1_1024/biases',786 'Conv2d_0_3x3_s2_1024/weights',787 'Conv2d_0_3x3_s2_1024/biases',788 'Conv2d_1_3x3_s2_1024/weights',789 'Conv2d_1_3x3_s2_1024/biases',790 'Conv2d_2_3x3_s2_1024/weights',791 'Conv2d_2_3x3_s2_1024/biases',792 'Conv2d_3_3x3_s2_1024/weights',793 'Conv2d_3_3x3_s2_1024/biases',794 'Conv2d_4_3x3_s2_1024/weights',795 'Conv2d_4_3x3_s2_1024/biases',796 ])797 init_op = tf.global_variables_initializer()798 with self.test_session() as sess:799 sess.run(init_op)800 sess.run(feature_maps)801 actual_variable_set = set(802 [var.op.name for var in tf.trainable_variables()])803 if replace_pool_with_conv:804 self.assertSetEqual(expected_conv_variables, actual_variable_set)805 else:806 self.assertSetEqual(expected_pool_variables, actual_variable_set)807if __name__ == '__main__':...
pif_calculator.py
Source:pif_calculator.py
1import collections2import math3import operator4import random5import itertools6import pandas as pd7import numpy as np8from piflib.data_util import calculate_distribution, complete_feature_priors9from piflib.entropy import create_conditional_entropy_table10def compute_cigs(dataframe,11 feature_priors={},12 feature_accuracies={},13 samples=None):14 """Compute the cell information gain (CIG) for all cells in the dataset.15 Find the risk (as KL divergence from prior) for all attributes.16 :param dataframe: a Pandas DataFrame object containing tabular data17 :param feature_priors: feature_priors are optional. It is a dictionary mapping the18 feature index to an assumed prior. If not provided, the prior for19 the feature is calculated from the global distribution.20 :param feature_accuracies: `feature_accuracies` maps the feature index to the accuracy of the21 feature. If not provided for a feature, it defaults to 1.22 :return: a Pandas DataFrame containing the CIG values. The CIG values are at the same index as their corresponding23 cell values in the input dataframe.24 """25 unknown_features = 126 dataset = dataframe.values27 num_features = len(dataset[0])28 assert all(len(row) == num_features for row in dataset)29 feature_priors = complete_feature_priors(dataframe, feature_priors)30 feature_accuracies = feature_accuracies.copy()31 for i in range(num_features):32 if i not in feature_accuracies:33 feature_accuracies[i] = 134 feature_counts = [0] * num_features35 feature_kls = [[0] * len(dataset) for _ in range(num_features)]36 for is_ in sample_is(num_features, unknown_features, samples):37 feature_kls_this = find_kls_for_features(38 dataset,39 is_,40 feature_priors,41 feature_accuracies)42 for i, feature_kl in zip(is_, feature_kls_this):43 feature_kl_previous = feature_kls[i]44 feature_kls[i] = tuple(map(45 operator.add, feature_kl, feature_kl_previous))46 for i in is_:47 feature_counts[i] += 148 for i, denom in enumerate(feature_counts):49 feature_kls[i] = tuple(map(50 operator.truediv,51 feature_kls[i],52 itertools.repeat(denom)))53 return pd.DataFrame(list(zip(*feature_kls)), columns=dataframe.columns)54def compute_weighted_cigs(dataframe, feature_priors={}, feature_accuracies={}):55 """Compute the Weighted Cell Information Gain (wCIG) for all cells in the dataset.56 Find the risk (as KL divergence from prior) for all attributes.57 :param dataframe: a Pandas DataFrame object containing tabular data58 :param feature_priors: feature_priors are optional. It is a dictionary mapping the59 feature index to an assumed prior. If not provided, the prior for60 the feature is calculated from the global distribution.61 :param feature_accuracies: `feature_accuracies` maps the feature index to the accuracy of the62 feature. If not provided for a feature, it defaults to 1.63 :return: a Pandas DataFrame containing the wCIG values. The wCIG values are at the same index as their64 corresponding cell values in the input dataframe.65 """66 cigs = compute_cigs(dataframe, feature_priors=feature_priors, feature_accuracies=feature_accuracies)67 cond_entropy = create_conditional_entropy_table(dataframe)68 weights = cond_entropy['H(X|Y)'].values / cond_entropy['H(X)']69 weights = np.nan_to_num(weights)70 return (cigs * weights).round(2)71def compute_csfs(df, feature_priors={}, feature_accuracies={}):72 """Compute the Cell Surprise Factor (CSF) for all cells in the dataset.73 The CSF id defined as the change in probability for a cell value between the prior and the posterior distribution.74 :param dataframe: a Pandas DataFrame object containing tabular data75 :param feature_priors: feature_priors are optional. It is a dictionary mapping the76 feature index to an assumed prior. If not provided, the prior for77 the feature is calculated from the global distribution.78 :param feature_accuracies: `feature_accuracies` maps the feature index to the accuracy of the79 feature. If not provided for a feature, it defaults to 1.80 :return: a Pandas DataFrame containing the CSF values. The CSF values are at the same index as their81 corresponding cell values in the input dataframe.82 """83 dataset = df.values84 num_features = len(dataset[0])85 # compute priors86 feature_priors = complete_feature_priors(df, feature_priors)87 feature_accuracies = feature_accuracies.copy()88 for i in range(num_features):89 if i not in feature_accuracies:90 feature_accuracies[i] = 191 feature_csfs = [[0] * len(dataset) for _ in range(num_features)]92 for is_ in sample_is(num_features, 1, None):93 feature_csfs[is_[0]] = apply_to_posterior_and_prior(94 dataset,95 is_,96 feature_priors,97 feature_accuracies,98 calculate_prob_change)99 return pd.DataFrame(list(zip(*feature_csfs)), columns=df.columns)100def compute_pif(cigs, percentile):101 """ compute the PIF.102 The PIF is defined as the n-th percentile of the individual RIG values. Or in other words, the RIG of n percent of103 the entities in the dataset does not exceed the PIF value.104 RIG stands for row information gain. It represents the overall information gain for an entity in the dataset. The105 RIG is computed by summing the CIG values of an entity.106 The percentile value can be chosen between 0 and 100. 100 will return the maximum RIG value. Often, the RIG values107 from a long tail distribution with few high value outliers. Choosing a percentile value lower than 100 will ignore108 (some of) the highest values. If ignoring the risk of some entities in the dataset fits within your risk framework,109 then specifying a percentile value of less than 100 will make the PIF value less susceptible to RIG outliers.110 :param cigs: The CIG values of the dataset (see the compute_cigs function in this module)111 :param percentile: Which percentile of RIG values should be included in the PIF.112 :returns: the PIF_percentile value of the given CIGs113 """114 rigs = cigs.sum(axis=1)115 pif = np.percentile(rigs, percentile)116 return pif117def compute_posterior_distributions(feature, df):118 known_features = tuple(col_name for col_name in df.columns if col_name != feature)119 bucket = collections.defaultdict(list)120 bucket_map = []121 for idx, row in df.iterrows():122 key = tuple(row[known_feature] for known_feature in known_features)123 bucket[key].append(row[feature])124 bucket_map.append(key)125 bucket_distributions = {key: calculate_distribution(el_bucket) for key, el_bucket in bucket.items()}126 feature_vals = df[feature].unique()127 dists = {}128 for key, distribution in bucket_distributions.items():129 dists[str(key)] = [distribution.get(feature_val, 0) for feature_val in feature_vals]130 return dists, feature_vals131def binom(n, r):132 """ return binomial coefficient: n choose k"""133 return math.factorial(n) // math.factorial(n - r) // math.factorial(r)134def sample_is(n, r, samples):135 if samples is None:136 yield from itertools.combinations(range(n), r)137 else:138 total_combinations = binom(n, r)139 if samples > total_combinations:140 raise ValueError('more samples than combinations')141 if samples >= total_combinations >> 1:142 all_combinations = list(itertools.combinations(range(n), r))143 random.shuffle(all_combinations)144 num_produced = 0145 feature_produced = [False] * n146 for i, comb in enumerate(all_combinations):147 if num_produced >= samples:148 break149 if all(map(feature_produced.__getitem__, comb)):150 continue151 for j in comb:152 feature_produced[j] = True153 num_produced += 1154 all_combinations[i] = None155 yield comb156 for comb in all_combinations:157 if num_produced >= samples:158 break159 if comb is not None:160 yield comb161 else:162 already_produced = set()163 feature_produced = [False] * n164 while len(already_produced) < samples:165 comb = random.sample(range(n), r)166 comb = tuple(sorted(comb))167 if (comb not in already_produced168 and (all(already_produced)169 or not all(map(already_produced.__getitem__,170 comb)))):171 already_produced.add(comb)172 for i in comb:173 feature_produced[i] = True174 yield comb175def apply_to_posterior_and_prior(dataset, feature_idx, prior_distributions, accuracies, fun):176 num_features = len(dataset[0])177 assert all(len(row) == num_features for row in dataset)178 feature_idx = feature_idx[0]179 buckets = collections.defaultdict(list)180 bucket_map = []181 for row in dataset:182 key = tuple(row[i] for i in range(num_features) if not i == feature_idx)183 buckets[key].append(row[feature_idx])184 bucket_map.append((key, row[feature_idx]))185 bucket_values = {186 key: fun(187 calculate_distribution(bucket,188 accuracy=accuracies[feature_idx],189 feature_distribution=prior_distributions[feature_idx]),190 prior_distributions[feature_idx])191 for key, bucket in buckets.items()}192 return [bucket_values[post_key][val] for post_key, val in bucket_map]193def find_kls_for_features(dataset, feature_is, feature_distributions, accuracies):194 """Find the KL divergence of feature values against the prior.195 We find the true distribution of the features taking into account196 the accuracy. We then compute the KL divergence.197 """198 num_features = len(dataset[0])199 assert all(len(row) == num_features for row in dataset)200 # one bucket per set of 'known' features201 buckets = [collections.defaultdict(list) for _ in range(len(feature_is))]202 bucket_map = [[] for _ in range(len(feature_is))]203 for row in dataset:204 key = tuple(row[i] for i in range(num_features) if i not in feature_is)205 for i, j in enumerate(feature_is):206 buckets[i][key].append(row[j])207 bucket_map[i].append(key)208 bucket_kls = [209 {210 key: calculate_kl(211 calculate_distribution(bucket,212 accuracy=accuracies[feature_is[i]],213 feature_distribution=feature_distributions[feature_is[i]]),214 feature_distributions[feature_is[i]])215 for key, bucket in feature_buckets.items()}216 for i, feature_buckets in enumerate(buckets)]217 return [tuple(map(bucket_kls[i].__getitem__, bucket_map[i]))218 for i in range(len(feature_is))]219def calculate_kl(p, q):220 """Calculate D_KL(P || Q) (the KL-divergence) in bits.221 D_KL(P || Q) is the `information gained when one revises one's222 beliefs from the prior probability distribution Q to the posterior223 probability distribution P`. (Wikipedia, KullbackâLeibler224 divergence)225 `p` and `q` are both dictionaries mapping some hashable to a number.226 It is assumed that they are both normalised: their values should add227 up to 0. `q` must not have any 0 values unless the corresponding `p`228 value is also 0.229 """230 return sum(pk * math.log2(pk / q[k])231 for k, pk in p.items()232 if pk > 0)233def calculate_prob_change(p, q):234 """ calculate the change in probability for each element of the posterior compared to the prior"""...
trip_quad.py
Source:trip_quad.py
1from __future__ import division2import pickle3import numpy as np4import cv25import math6from sys import argv7from copy import *8from ProjectUtils import *9from GetDataPath import *10from testvgg import TestVgg11sys.path.insert(0, './')12sys.path.append('/home/xuyangf/project/ML_deliverables/Siamese_iclr17_tf-master/src/')13import network as vgg14from feature_extractor import FeatureExtractor15from utils import *16# import importlib17# importlib.reload(sys)18reload(sys) 19sys.setdefaultencoding('utf8')20cat=argv[1]21cluster_num = 25622myimage_path=LoadImage(cat)23image_path=[]24for mypath in myimage_path:25 myimg=cv2.imread(mypath, cv2.IMREAD_UNCHANGED)26 if(max(myimg.shape[0],myimg.shape[1])>100):27 image_path.append(mypath)28img_num=len(image_path)29layer_name = 'pool3'30file_path = '/data2/xuyangf/OcclusionProject/NaiveVersion/feature/feature3/L3Feature'+cat31#cluster_file = '/data2/xuyangf/OcclusionProject/NaiveVersion/cluster/clusterL3/vgg16_'+cat+'_K'+str(cluster_num)+'.pickle'32prun_file = '/data2/xuyangf/OcclusionProject/NaiveVersion/prunning/prunL3/dictionary_'+cat+'.pickle'33trip_save_file='/data2/xuyangf/OcclusionProject/NaiveVersion/vc_combinescore/3vc/cat'+cat34quad_save_file='/data2/xuyangf/OcclusionProject/NaiveVersion/vc_combinescore/4vc/cat'+cat35print('loading data...')36# number of files to read in37file_num = 1038maximg_cnt=2000039fname = file_path+str(0)+'.npz'40ff = np.load(fname)41feat_dim = ff['res'].shape[0]42img_cnt = ff['res'].shape[1]43oldimg_index=044originimage=[]45feat_set = np.zeros((feat_dim, maximg_cnt*file_num))46feat_set[:,0:img_cnt] = ff['res']47originimage+=list(ff['originpath'])48loc_dim = ff['loc_set'].shape[1]49print(loc_dim)50loc_set = np.zeros((maximg_cnt*file_num, loc_dim))51loc_set[0:img_cnt,:] = ff['loc_set']52#img_dim = ff['img_set'].shape[1:]53#img_set = np.zeros([maximg_cnt*file_num]+list(img_dim))54#img_set[0:img_cnt] = ff['img_set']55oldimg_index+=img_cnt56for ii in range(1,file_num):57 print(ii)58 fname = file_path+str(ii)+'.npz'59 ff = np.load(fname)60 originimage+=list(ff['originpath'])61 img_cnt=ff['res'].shape[1]62 print(img_cnt)63 feat_set[:,oldimg_index:(oldimg_index + img_cnt)] = ff['res']64 loc_set[oldimg_index:(oldimg_index + img_cnt),:] = ff['loc_set']65 #img_set[oldimg_index:(oldimg_index + img_cnt)] = ff['img_set']66 oldimg_index+=img_cnt67feat_set=feat_set[:,:oldimg_index]68#img_set=img_set[:oldimg_index]69loc_set=loc_set[:oldimg_index,:]70print('all feat_set')71print(feat_set.shape)72print('all img_set')73#print(img_set.shape)74#assert(len(originimage)==len(img_set))75with open(prun_file, 'rb') as fh:76 assignment, centers, _, norm = pickle.load(fh)77print('load finish')78predictor=TestVgg()79# def GetPossDecrease(original_img,occluded_img,category):80# originalp=predictor.getprob(original_img,category)81# occludedp=predictor.getprob(occluded_img,category)82# print('diff')83# print(originalp)84# print(occludedp)85# #0.3 magic number86# if(originalp<0.3):87# return -188# if(originalp<occludedp):89# return -190# return float(originalp-occludedp)/originalp91def GetPossDecrease(original_feature,occluded_feature,category):92 originalp=predictor.feature2result(original_feature,category)93 occludedp=predictor.feature2result(occluded_feature,category)94 print('diff')95 print(originalp)96 print(occludedp)97 #0.3 magic number98 if(originalp<0.3):99 return -1100 # if(originalp<occludedp):101 # return -1102 return float(originalp-occludedp)/originalp103def disttresh(input_index,cluster_center):104 thresh1=0.8105 temp_feat=feat_set[:,input_index]106 error = np.sum((temp_feat.T - cluster_center)**2, 1)107 sort_idx = np.argsort(error)108 return input_index[sort_idx[:int(thresh1*len(sort_idx))]]109fname ='/data2/xuyangf/OcclusionProject/NaiveVersion/vc_combinescore/2vc/cat'+str(cat)+'.npz'110ff=np.load(fname)111img_vc_avg=ff['vc_score']112vc_name=ff['vc']113vc_num=len(img_vc_avg)114img_vc_avg=np.array(img_vc_avg)115vc_name=np.array(vc_name)116b=np.argsort(-img_vc_avg)117trip_img_vc_score=[]118trip_img_vc=[]119quad_img_vc_score=[]120quad_img_vc=[]121topvc=b[:25]122for indexk in topvc:123 thename=vc_name[indexk]124 k1=int(thename[0])125 k2=int(thename[1])126 for indexkk in topvc:127 if indexk<=indexkk:128 continue129 thename2=vc_name[indexkk]130 k3=int(thename2[0])131 k4=int(thename2[1])132 myk=[]133 myk.append(k1)134 myk.append(k2)135 myk.append(k3)136 myk.append(k4)137 myk=set(myk)138 myk=list(myk)139 if len(myk)==3:140 trip_img_vc.append(str(myk[0])+'_'+str(myk[1])+'_'+str(myk[2]))141 else:142 quad_img_vc.append(str(myk[0])+'_'+str(myk[1])+'_'+str(myk[2])+'_'+str(myk[3]))143trip_img_vc=list(set(trip_img_vc))144quad_img_vc=list(set(quad_img_vc))145for s in trip_img_vc:146 dot1=s.find('_')147 k1=int(s[:dot1])148 s2=s[dot1+1:]149 dot2=s2.find('_')150 k2=int(s2[:dot2])151 s3=s2[dot2+1:]152 k3=int(s3)153 target1=centers[k1]154 index1=np.where(assignment==k1)[0]155 index1=disttresh(index1,target1)156 target2=centers[k2]157 index2=np.where(assignment==k2)[0]158 index2=disttresh(index2,target2)159 target3=centers[k3]160 index3=np.where(assignment==k3)[0]161 index3=disttresh(index3,target3)162 myscore=[]163 for n in range(0,img_num):164 myindex1=[]165 for i in range(len(index1)):166 if image_path[n]==originimage[index1[i]]:167 myindex1.append(index1[i])168 #myindex=OnlyTheClosest(myindex,target), or other preprocessing method169 myindex2=[]170 for i in range(len(index2)):171 if image_path[n]==originimage[index2[i]]:172 myindex2.append(index2[i])173 myindex3=[]174 for i in range(len(index3)):175 if image_path[n]==originimage[index3[i]]:176 myindex3.append(index3[i])177 if len(myindex1)==0:178 continue179 if len(myindex2)==0:180 continue181 if len(myindex3)==0:182 continue183 myindex=myindex1+myindex2+myindex3184 original_img=cv2.imread(image_path[n], cv2.IMREAD_UNCHANGED)185 original_feature=predictor.img2feature(original_img)186 occluded_feature=deepcopy(original_feature)187 for i in range(len(myindex)):188 fhi=int(loc_set[myindex[i]][7])189 fwi=int(loc_set[myindex[i]][8])190 191 # Gause occlusion192 occluded_feature[0][fhi][fwi]=0193 occluded_feature[0][fhi-1][fwi]=0.25*occluded_feature[0][fhi-1][fwi]194 occluded_feature[0][fhi][fwi-1]=0.25*occluded_feature[0][fhi][fwi-1]195 occluded_feature[0][fhi+1][fwi]=0.25*occluded_feature[0][fhi+1][fwi]196 occluded_feature[0][fhi][fwi+1]=0.25*occluded_feature[0][fhi][fwi+1]197 occluded_feature[0][fhi-1][fwi-1]=0.375*occluded_feature[0][fhi-1][fwi-1]198 occluded_feature[0][fhi+1][fwi-1]=0.375*occluded_feature[0][fhi+1][fwi-1]199 occluded_feature[0][fhi+1][fwi+1]=0.375*occluded_feature[0][fhi+1][fwi+1]200 occluded_feature[0][fhi-1][fwi+1]=0.375*occluded_feature[0][fhi-1][fwi+1]201 202 occluded_feature[0][fhi+2][fwi]=0.625*occluded_feature[0][fhi+2][fwi]203 occluded_feature[0][fhi-2][fwi]=0.625*occluded_feature[0][fhi-2][fwi]204 occluded_feature[0][fhi][fwi-2]=0.625*occluded_feature[0][fhi][fwi-2]205 occluded_feature[0][fhi][fwi+2]=0.625*occluded_feature[0][fhi][fwi+2]206 occluded_feature[0][fhi-1][fwi-2]=0.75*occluded_feature[0][fhi-1][fwi-2]207 occluded_feature[0][fhi-1][fwi+2]=0.75*occluded_feature[0][fhi-1][fwi+2]208 occluded_feature[0][fhi+1][fwi-2]=0.75*occluded_feature[0][fhi+1][fwi-2]209 occluded_feature[0][fhi+1][fwi+2]=0.75*occluded_feature[0][fhi+1][fwi+2]210 occluded_feature[0][fhi-2][fwi-1]=0.75*occluded_feature[0][fhi-2][fwi-1]211 occluded_feature[0][fhi-2][fwi+1]=0.75*occluded_feature[0][fhi-2][fwi+1]212 occluded_feature[0][fhi+2][fwi-1]=0.75*occluded_feature[0][fhi+2][fwi-1]213 occluded_feature[0][fhi+2][fwi+1]=0.75*occluded_feature[0][fhi+2][fwi+1]214 occluded_feature[0][fhi-2][fwi-2]=0.875*occluded_feature[0][fhi-21][fwi-2]215 occluded_feature[0][fhi-2][fwi+2]=0.875*occluded_feature[0][fhi-2][fwi+2]216 occluded_feature[0][fhi+2][fwi-2]=0.875*occluded_feature[0][fhi+2][fwi-2]217 occluded_feature[0][fhi+2][fwi+2]=0.875*occluded_feature[0][fhi+2][fwi+2] 218 # print(hi)219 # print(wi)220 # print(patch_size)221 drop=GetPossDecrease(original_feature,occluded_feature,int(cat))222 if drop!=-1:223 myscore.append(drop)224 trip_img_vc_score.append(float(np.sum(myscore))/img_num)225np.savez(trip_save_file,vc_score=trip_img_vc_score,vc=trip_img_vc)226for s in quad_img_vc:227 dot1=s.find('_')228 k1=int(s[:dot1])229 s2=s[dot1+1:]230 dot2=s2.find('_')231 k2=int(s2[:dot2])232 s3=s2[dot2+1:]233 dot3=s3.find('_')234 k3=int(s3[:dot3])235 s4=s3[dot3+1:]236 k4=int(s4)237 target1=centers[k1]238 index1=np.where(assignment==k1)[0]239 index1=disttresh(index1,target1)240 target2=centers[k2]241 index2=np.where(assignment==k2)[0]242 index2=disttresh(index2,target2)243 target3=centers[k3]244 index3=np.where(assignment==k3)[0]245 index3=disttresh(index3,target3)246 target4=centers[k4]247 index4=np.where(assignment==k4)[0]248 index4=disttresh(index4,target4)249 myscore=[]250 for n in range(0,img_num):251 myindex1=[]252 for i in range(len(index1)):253 if image_path[n]==originimage[index1[i]]:254 myindex1.append(index1[i])255 #myindex=OnlyTheClosest(myindex,target), or other preprocessing method256 myindex2=[]257 for i in range(len(index2)):258 if image_path[n]==originimage[index2[i]]:259 myindex2.append(index2[i])260 myindex3=[]261 for i in range(len(index3)):262 if image_path[n]==originimage[index3[i]]:263 myindex3.append(index3[i])264 myindex4=[]265 for i in range(len(index4)):266 if image_path[n]==originimage[index4[i]]:267 myindex4.append(index4[i])268 if len(myindex1)==0:269 continue270 if len(myindex2)==0:271 continue272 if len(myindex3)==0:273 continue274 if len(myindex4)==0:275 continue276 myindex=myindex1+myindex2+myindex3+myindex4277 original_img=cv2.imread(image_path[n], cv2.IMREAD_UNCHANGED)278 original_feature=predictor.img2feature(original_img)279 occluded_feature=deepcopy(original_feature)280 for i in range(len(myindex)):281 fhi=int(loc_set[myindex[i]][7])282 fwi=int(loc_set[myindex[i]][8])283 284 # Gause occlusion285 occluded_feature[0][fhi][fwi]=0286 occluded_feature[0][fhi-1][fwi]=0.25*occluded_feature[0][fhi-1][fwi]287 occluded_feature[0][fhi][fwi-1]=0.25*occluded_feature[0][fhi][fwi-1]288 occluded_feature[0][fhi+1][fwi]=0.25*occluded_feature[0][fhi+1][fwi]289 occluded_feature[0][fhi][fwi+1]=0.25*occluded_feature[0][fhi][fwi+1]290 occluded_feature[0][fhi-1][fwi-1]=0.375*occluded_feature[0][fhi-1][fwi-1]291 occluded_feature[0][fhi+1][fwi-1]=0.375*occluded_feature[0][fhi+1][fwi-1]292 occluded_feature[0][fhi+1][fwi+1]=0.375*occluded_feature[0][fhi+1][fwi+1]293 occluded_feature[0][fhi-1][fwi+1]=0.375*occluded_feature[0][fhi-1][fwi+1]294 295 occluded_feature[0][fhi+2][fwi]=0.625*occluded_feature[0][fhi+2][fwi]296 occluded_feature[0][fhi-2][fwi]=0.625*occluded_feature[0][fhi-2][fwi]297 occluded_feature[0][fhi][fwi-2]=0.625*occluded_feature[0][fhi][fwi-2]298 occluded_feature[0][fhi][fwi+2]=0.625*occluded_feature[0][fhi][fwi+2]299 occluded_feature[0][fhi-1][fwi-2]=0.75*occluded_feature[0][fhi-1][fwi-2]300 occluded_feature[0][fhi-1][fwi+2]=0.75*occluded_feature[0][fhi-1][fwi+2]301 occluded_feature[0][fhi+1][fwi-2]=0.75*occluded_feature[0][fhi+1][fwi-2]302 occluded_feature[0][fhi+1][fwi+2]=0.75*occluded_feature[0][fhi+1][fwi+2]303 occluded_feature[0][fhi-2][fwi-1]=0.75*occluded_feature[0][fhi-2][fwi-1]304 occluded_feature[0][fhi-2][fwi+1]=0.75*occluded_feature[0][fhi-2][fwi+1]305 occluded_feature[0][fhi+2][fwi-1]=0.75*occluded_feature[0][fhi+2][fwi-1]306 occluded_feature[0][fhi+2][fwi+1]=0.75*occluded_feature[0][fhi+2][fwi+1]307 occluded_feature[0][fhi-2][fwi-2]=0.875*occluded_feature[0][fhi-21][fwi-2]308 occluded_feature[0][fhi-2][fwi+2]=0.875*occluded_feature[0][fhi-2][fwi+2]309 occluded_feature[0][fhi+2][fwi-2]=0.875*occluded_feature[0][fhi+2][fwi-2]310 occluded_feature[0][fhi+2][fwi+2]=0.875*occluded_feature[0][fhi+2][fwi+2] 311 # print(hi)312 # print(wi)313 # print(patch_size)314 drop=GetPossDecrease(original_feature,occluded_feature,int(cat))315 if drop!=-1:316 myscore.append(drop)317 quad_img_vc_score.append(float(np.sum(myscore))/img_num)...
DragFeature.js
Source:DragFeature.js
...136 }, this.dragCallbacks), {137 documentDrag: this.documentDrag138 }139 ),140 feature: new OpenLayers.Handler.Feature(141 this, this.layer, OpenLayers.Util.extend({142 // 'click' and 'clickout' callback are for the mobile143 // support: no 'over' or 'out' in touch based browsers.144 click: this.clickFeature,145 clickout: this.clickoutFeature,146 over: this.overFeature,147 out: this.outFeature148 }, this.featureCallbacks),149 {geometryTypes: this.geometryTypes}150 )151 };152 },153 /**154 * Method: clickFeature155 * Called when the feature handler detects a click-in on a feature.156 *157 * Parameters:158 * feature - {<OpenLayers.Feature.Vector>}159 */160 clickFeature: function(feature) {161 if (this.handlers.feature.touch && !this.over && this.overFeature(feature)) {162 this.handlers.drag.dragstart(this.handlers.feature.evt);163 // to let the events propagate to the feature handler (click callback)164 this.handlers.drag.stopDown = false;165 }166 },167 /**168 * Method: clickoutFeature169 * Called when the feature handler detects a click-out on a feature.170 *171 * Parameters:172 * feature - {<OpenLayers.Feature.Vector>}173 */174 clickoutFeature: function(feature) {175 if (this.handlers.feature.touch && this.over) {176 this.outFeature(feature);177 this.handlers.drag.stopDown = true;178 }179 },180 /**181 * APIMethod: destroy182 * Take care of things that are not handled in superclass183 */184 destroy: function() {185 this.layer = null;186 OpenLayers.Control.prototype.destroy.apply(this, []);187 },188 /**189 * APIMethod: activate190 * Activate the control and the feature handler.191 * 192 * Returns:193 * {Boolean} Successfully activated the control and feature handler.194 */195 activate: function() {196 return (this.handlers.feature.activate() &&197 OpenLayers.Control.prototype.activate.apply(this, arguments));198 },199 /**200 * APIMethod: deactivate201 * Deactivate the control and all handlers.202 * 203 * Returns:204 * {Boolean} Successfully deactivated the control.205 */206 deactivate: function() {207 // the return from the handlers is unimportant in this case208 this.handlers.drag.deactivate();209 this.handlers.feature.deactivate();210 this.feature = null;211 this.dragging = false;212 this.lastPixel = null;213 OpenLayers.Element.removeClass(214 this.map.viewPortDiv, this.displayClass + "Over"215 );216 return OpenLayers.Control.prototype.deactivate.apply(this, arguments);217 },218 /**219 * Method: overFeature220 * Called when the feature handler detects a mouse-over on a feature.221 * This activates the drag handler.222 *223 * Parameters:224 * feature - {<OpenLayers.Feature.Vector>} The selected feature.225 *226 * Returns:227 * {Boolean} Successfully activated the drag handler.228 */229 overFeature: function(feature) {230 var activated = false;231 if(!this.handlers.drag.dragging) {232 this.feature = feature;233 this.handlers.drag.activate();234 activated = true;235 this.over = true;236 OpenLayers.Element.addClass(this.map.viewPortDiv, this.displayClass + "Over");237 this.onEnter(feature);238 } else {239 if(this.feature.id == feature.id) {240 this.over = true;241 } else {242 this.over = false;243 }244 }245 return activated;246 },247 /**248 * Method: downFeature249 * Called when the drag handler detects a mouse-down.250 *251 * Parameters:252 * pixel - {<OpenLayers.Pixel>} Location of the mouse event.253 */254 downFeature: function(pixel) {255 this.lastPixel = pixel;256 this.onStart(this.feature, pixel);257 },258 /**259 * Method: moveFeature260 * Called when the drag handler detects a mouse-move. Also calls the261 * optional onDrag method.262 * 263 * Parameters:264 * pixel - {<OpenLayers.Pixel>} Location of the mouse event.265 */266 moveFeature: function(pixel) {267 var res = this.map.getResolution();268 this.feature.geometry.move(res * (pixel.x - this.lastPixel.x),269 res * (this.lastPixel.y - pixel.y));270 this.layer.drawFeature(this.feature);271 this.lastPixel = pixel;272 this.onDrag(this.feature, pixel);273 },274 /**275 * Method: upFeature276 * Called when the drag handler detects a mouse-up.277 * 278 * Parameters:279 * pixel - {<OpenLayers.Pixel>} Location of the mouse event.280 */281 upFeature: function(pixel) {282 if(!this.over) {283 this.handlers.drag.deactivate();284 }...
ToolboxView.js
Source:ToolboxView.js
...46 var Feature = featureManager.get(featureName);47 if (!Feature) {48 return;49 }50 feature = new Feature(featureModel);51 }52 features[featureName] = feature;53 }54 else {55 feature = features[oldName];56 // If feature does not exsit.57 if (!feature) {58 return;59 }60 feature.model = featureModel;61 }62 if (!featureName && oldName) {63 feature.dispose && feature.dispose(ecModel, api);64 return;...
Using AI Code Generation
1var ladle = require('ladle');2var feature = ladle.feature();3var scenario = feature.scenario;4var given = scenario.given;5var when = scenario.when;6var then = scenario.then;7feature.load(__dirname + '/test.feature');8feature.run(function(err, summary) {9 if (err) {10 console.log('Error: ' + err);11 } else {12 console.log('Feature: ' + summary.feature);13 console.log(' Scenarios: ' + summary.scenarios);14 console.log(' Passed: ' + summary.passed);15 console.log(' Failed: ' + summary.failed);16 }17});18var ladle = require('ladle');19var feature = ladle.feature();20var scenario = feature.scenario;21var given = scenario.given;22var when = scenario.when;23var then = scenario.then;24feature.load(__dirname + '/test.feature');25feature.run(function(err, summary) {26 if (err) {27 console.log('Error: ' + err);28 } else {29 console.log('Feature: ' + summary.feature);30 console.log(' Scenarios: ' + summary.scenarios);31 console.log(' Passed: ' + summary.passed);32 console.log(' Failed: ' + summary.failed);33 }34});35var ladle = require('ladle');36var feature = ladle.feature();37var scenario = feature.scenario;38var given = scenario.given;39var when = scenario.when;40var then = scenario.then;41feature.load(__dirname + '/test.feature');42feature.run(function(err, summary) {43 if (err) {44 console.log('Error: ' + err);45 } else {46 console.log('Feature: ' + summary.feature);47 console.log(' Scenarios: ' + summary.scenarios);48 console.log(' Passed: ' + summary.passed);49 console.log(' Failed:
Using AI Code Generation
1var ladle = require('ladle');2var feature = ladle.Feature('test.feature', 'test.js');3feature.run();4var ladle = require('ladle');5var feature = ladle.Feature('test.feature', 'test.js');6feature.run();7module.exports = function() {8 this.Given(/^I have a test$/, function (callback) {9 callback(null, 'pending');10 });11 this.When(/^I run the test$/, function (callback) {12 callback(null, 'pending');13 });14 this.Then(/^I get the result$/, function (callback) {15 callback(null, 'pending');16 });17};
Using AI Code Generation
1var ladle = require('ladle');2var feature = ladle.Feature('test.feature');3feature.scenario('test', function(scenario) {4 scenario.Given(/^I have entered (\d+) into the calculator$/, function (arg1, callback) {5 callback(null, 'pending');6 });7 scenario.When(/^I press add$/, function (callback) {8 callback(null, 'pending');9 });10 scenario.Then(/^the result should be (\d+) on the screen$/, function (arg1, callback) {11 callback(null, 'pending');12 });13});141 scenario (1 undefined)153 steps (3 undefined)16module.exports = function () {17 this.Given(/^I have entered (\d+) into the calculator$/, function (arg1, callback) {18 callback(null, 'pending');19 });20 this.When(/^I press add$/, function (callback) {21 callback(null, 'pending');22 });23 this.Then(/^the result should be (\d+) on the screen$/, function (arg1, callback) {24 callback(null, 'pending');25 });26};
Using AI Code Generation
1var ladle = require('ladle');2var feature = ladle.Feature('test.feature');3feature.run();4var ladle = require('ladle');5var feature = ladle.Feature('test.feature');6feature.run();7var ladle = require('ladle');8var feature = ladle.Feature('test.feature');9feature.run();10var ladle = require('ladle');11var feature = ladle.Feature('test.feature');12feature.run();13var ladle = require('ladle');14var feature = ladle.Feature('test.feature');15feature.run();16var ladle = require('ladle');17var feature = ladle.Feature('test.feature');18feature.run();19var ladle = require('ladle');20var feature = ladle.Feature('test.feature');21feature.run();22var ladle = require('ladle');
Using AI Code Generation
1var ladle = require('ladle');2var feature = ladle.feature('test.feature');3feature.scenario('test scenario', function(scenario) {4 scenario.given('I have a step', function(next) {5 next();6 });7 scenario.when('I have another step', function(next) {8 next();9 });10 scenario.then('I have a final step', function(next) {11 next();12 });13});14 3 steps (3 passed)15feature.scenario('test scenario', function(scenario) {16 scenario.given('I have a step', function(next) {17 next();18 });19 scenario.when('I have another step', function(next) {20 next();21 });22 scenario.then('I have a final step', function(next) {23 next();24 });25 scenario.and('I have a step that takes a long time', function
Using AI Code Generation
1var ladle = require('ladle');2var feature = ladle.feature('test.feature');3var world = feature.world;4feature.scenario('Test scenario', function(scenario) {5 scenario.given('I am on the home page', function() {6 console.log('I am on the home page');7 });8 });9 scenario.when('I search for Ladle', function() {10 world.browser.fill('q', 'ladle').pressButton('btnG', function() {11 console.log('I search for Ladle');12 });13 });14 scenario.then('I should see Ladle on the results page', function() {15 world.browser.assert.text('h3.r', 'Ladle');16 console.log('I should see Ladle on the results page');17 });18});
Using AI Code Generation
1const ladle = require('ladle');2const feature = ladle.feature('test.feature');3feature.scenario('test', () => {4 feature.Given('test', () => {5 });6 feature.Then('test', () => {7 });8});9const ladle = require('ladle');10ladle.feature('test.feature').scenario('test', () => {11 feature.Given('test', () => {12 });13 feature.Then('test', () => {14 });15});16const ladle = require('ladle');17ladle.feature('test.feature').scenario('test', () => {18 feature.Given('test', () => {19 });20 feature.Then('test', () => {21 });22});23const ladle = require('ladle');24ladle.feature('test.feature').scenario('test', () => {25 feature.Given('test', () => {26 });27 feature.Then('test', () => {28 });29});30const ladle = require('ladle');31ladle.feature('test.feature').scenario('test', () => {32 feature.Given('test', () => {33 });34 feature.Then('test', () => {35 });36});37const ladle = require('ladle
Using AI Code Generation
1var ladle = require('ladle');2var feature = ladle.Feature();3var feature2 = ladle.Feature();4feature2.addStep('I go to google', function (callback) {5});6feature2.addStep('I search for ladle', function (callback) {7 this.browser.fill('q', 'ladle').pressButton('btnG', callback);8});9feature2.addStep('I should see ladle in the results', function (callback) {10 assert.ok(this.browser.success);11 assert.ok(this.browser.text('body').indexOf('ladle') !== -1);12 callback();13});14feature2.scenario('I can search google', function (feature) {15 feature.startingWith('I go to google');16 feature.when('I search for ladle');17 feature.then('I should see ladle in the results');18});19feature.addFeature(feature2);20feature.run(function (err) {21 if (err) {22 console.log(err);23 }24});25var ladle = require('ladle');26var feature = ladle.Feature();27feature.addStep('I go to google', function (callback) {28});29feature.addStep('I search for ladle', function (callback) {30 this.browser.fill('q', 'ladle').pressButton('btnG', callback);31});32feature.addStep('I should see ladle in the results', function (callback) {33 assert.ok(this.browser.success);34 assert.ok(this.browser.text('body').indexOf('ladle') !== -1);35 callback();36});37feature.scenario('I can search google', function (feature) {38 feature.startingWith('I go to google');39 feature.when('I search for ladle');40 feature.then('I should see ladle in the results');41});42feature.run(function (err) {43 if (err) {44 console.log(err);45 }46});
Using AI Code Generation
1var ladle = require('../index.js');2var feature = new ladle.Feature(ladle.Feature.CUCUMBER, 'test.feature');3feature.load(function(err, data) {4 console.log(data);5});6{ feature: 'Test Feature',7 [ { scenario: 'Test Scenario',8 examples: [] } ],9 tags: [] }10var ladle = require('../index.js');11var feature = new ladle.Feature(ladle.Feature.CUCUMBER, 'test.feature');12feature.load(function(err, data) {13 console.log(data);14});15{ feature: 'Test Feature',16 [ { scenario: 'Test Scenario',17 examples: [Object] } ],18 tags: [] }19var ladle = require('../index.js');20var feature = new ladle.Feature(ladle.Feature.CUCUMBER, 'test.feature');21feature.load(function(err, data) {22 console.log(data);23});24{ feature: 'Test Feature',25 [ { scenario: 'Test Scenario',26 examples: [Object] },27 { scenario: 'Test Scenario 2',28 examples: [] }
Automation Testing Tutorials
Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.
LambdaTest Learning Hubs:
- JUnit Tutorial
- TestNG Tutorial
- Webdriver Tutorial
- WebDriverIO Tutorial
- Protractor Tutorial
- Selenium 4 Tutorial
- Jenkins Tutorial
- NUnit Tutorial
- Jest Tutorial
- Playwright Tutorial
- Cypress Tutorial
- PyTest Tutorial
YouTube
You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.
Run ladle automation tests on LambdaTest cloud grid
Perform automation testing on 3000+ real desktop and mobile devices online.
Try LambdaTest Now !!
Get 100 minutes of automation test minutes FREE!!
Was this article helpful?
