| 54 | |
| 55 | |
| 56 | class Model(tp.ModelDescBase): |
| 57 | def inputs(self): |
| 58 | return [tf.TensorSpec((IMAGE_SIZE, IMAGE_SIZE, 3), tf.float32, 'image')] |
| 59 | |
| 60 | def build_graph(self, orig_image): |
| 61 | mean = tf.get_variable('resnet_v1_50/mean_rgb', shape=[3]) |
| 62 | with guided_relu(): |
| 63 | with slim.arg_scope(resnet_v1.resnet_arg_scope()): |
| 64 | image = tf.expand_dims(orig_image - mean, 0) |
| 65 | logits, _ = resnet_v1.resnet_v1_50(image, 1000, is_training=False) |
| 66 | saliency_map(logits, orig_image, name="saliency") |
| 67 | |
| 68 | |
| 69 | def run(model_path, image_path): |
no outgoing calls
no test coverage detected
searching dependent graphs…