Spaces:
Runtime error
Runtime error
File size: 439 Bytes
12d50ff |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from .utils.model_utils import setup_model
def get_latents(net, x, is_cars=False):
codes = net.encoder(x)
if net.opts.start_from_latent_avg:
if codes.ndim == 2:
codes = codes + net.latent_avg.repeat(codes.shape[0], 1, 1)[:, 0, :]
else:
codes = codes + net.latent_avg.repeat(codes.shape[0], 1, 1)
if codes.shape[1] == 18 and is_cars:
codes = codes[:, :16, :]
return codes
|