(
self, device, batch_size, relative, use_wildcard, num_threads=3, device_id=0, num_gpus=1
)
| 109 | |
| 110 | class ReshapeWithArgInput(Pipeline): |
| 111 | def __init__( |
| 112 | self, device, batch_size, relative, use_wildcard, num_threads=3, device_id=0, num_gpus=1 |
| 113 | ): |
| 114 | super(ReshapeWithArgInput, self).__init__( |
| 115 | batch_size, num_threads, device_id, seed=7865, exec_async=False, exec_pipelined=False |
| 116 | ) |
| 117 | self.device = device |
| 118 | self.input = ops.readers.Caffe( |
| 119 | path=caffe_db_folder, shard_id=device_id, num_shards=num_gpus |
| 120 | ) |
| 121 | self.resize = ops.Resize(device="cpu") |
| 122 | self.decode = ops.decoders.Image(device="cpu", output_type=types.RGB) |
| 123 | self.gen_shapes = ops.PythonFunction(function=MakeTallFunc(relative, use_wildcard)) |
| 124 | self.reshape = ops.Reshape(device=device) |
| 125 | self.relative = relative |
| 126 | |
| 127 | def define_graph(self): |
| 128 | jpegs, labels = self.input(name="Reader") |
nothing calls this directly
no test coverage detected