import numpy as np import torch import torch.nn as nn from torch.utils.data import DataLoader, Dataset from torchvision import models, transforms from PIL import Image from tqdm import tqdm

Xiuren photo is a style of photography that originated in China, characterized by:

#!/usr/bin/env python # -*- coding: utf-8 -*-

# ---- 6️⃣ Inference loop ---------------------------------------------------- with torch.no_grad(): for imgs, paths in tqdm(loader, desc="Extracting", unit="batch"): imgs = imgs.to(device, non_blocking=True) feats = model(imgs) # (B, 2048) if args.normalize: feats = torch.nn.functional.normalize(feats, p=2, dim=1) all_feats.append(feats.cpu().numpy()) all_paths.extend(paths)