MCPcopy Create free account
hub / github.com/NVIDIA/FastPhotoStyle / forward

Method forward

models.py:251–297  ·  view source on GitHub ↗
(self, x, pool1_idx=None, pool1_size=None, pool2_idx=None, pool2_size=None, pool3_idx=None,
                pool3_size=None)

Source from the content-addressed store, hash-verified

249 self.conv1_1 = nn.Conv2d(64, 3, 3, 1, 0)
250
251 def forward(self, x, pool1_idx=None, pool1_size=None, pool2_idx=None, pool2_size=None, pool3_idx=None,
252 pool3_size=None):
253 out = x
254
255 if self.level > 3:
256 out = self.pad4_1(out)
257 out = self.conv4_1(out)
258 out = self.relu4_1(out)
259 out = self.unpool3(out, pool3_idx, output_size=pool3_size)
260
261 out = self.pad3_4(out)
262 out = self.conv3_4(out)
263 out = self.relu3_4(out)
264
265 out = self.pad3_3(out)
266 out = self.conv3_3(out)
267 out = self.relu3_3(out)
268
269 out = self.pad3_2(out)
270 out = self.conv3_2(out)
271 out = self.relu3_2(out)
272
273 if self.level > 2:
274 out = self.pad3_1(out)
275 out = self.conv3_1(out)
276 out = self.relu3_1(out)
277 out = self.unpool2(out, pool2_idx, output_size=pool2_size)
278
279 out = self.pad2_2(out)
280 out = self.conv2_2(out)
281 out = self.relu2_2(out)
282
283 if self.level > 1:
284 out = self.pad2_1(out)
285 out = self.conv2_1(out)
286 out = self.relu2_1(out)
287 out = self.unpool1(out, pool1_idx, output_size=pool1_size)
288
289 out = self.pad1_2(out)
290 out = self.conv1_2(out)
291 out = self.relu1_2(out)
292
293 if self.level > 0:
294 out = self.pad1_1(out)
295 out = self.conv1_1(out)
296
297 return out

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected