Upload 3 files (#1)

- Upload 3 files (02370bc9cf48025cef1aaf47db071f4745dc8aaa)

efnet_step_86400.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d43954ae99774d23919f9b9ce2b39edf469ade3fae2496948b87e1b29079f8d0
+size 214488367

inference.py

@@ -0,0 +1,44 @@
+import time
+from PIL import Image
+from timm.data import resolve_data_config
+import torch
+from torchvision.transforms import transforms
+
+model = torch.load('path/to/model.pth')
+model.eval()
+
+config = resolve_data_config({}, model=model)
+transform = transforms.Compose([
+                transforms.Resize((224, 224)),
+                transforms.ToTensor(),
+                transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])  # Normalize image
+            ])
+
+
+with open("tags.txt", "r") as f:
+    categories = [s.strip() for s in f.readlines()]
+    categories=sorted(categories)
+
+
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
+images=["your_image_here.jpg", "your_second_image_here.jpg"]
+
+for item in images:
+    start = time.time()
+    img = Image.open(item).convert('RGB')
+    tensor = transform(img).unsqueeze(0).to(device) # transform and add batch dimension
+
+    with torch.no_grad():
+        out = model(tensor)
+    probabilities = torch.nn.functional.sigmoid(out[0])
+    print(probabilities.shape)
+
+
+    top10_prob, top10_catid = torch.topk(probabilities, 10)
+    for i in range(top10_prob.size(0)):
+        print(categories[top10_catid[i]], top10_prob[i].item())
+
+    end = time.time()
+    print(f'Executed in {end - start} seconds')
+    print("\n\n", end="")

tags.txt

@@ -0,0 +1,105 @@
+male
+female
+anthro
+breasts
+solo
+canine
+cock
+sex
+pussy
+pokemon
+ass
+cute
+thighs
+feet
+paws
+blush
+femboy
+muscles
+size_difference
+pony
+equine
+sfw
+nsfw
+mlp
+public
+pov
+femdom
+scales
+thick_thighs
+genitals
+mammal
+bodily_fluids
+duo
+fur
+clothing
+canid
+hair
+genital_fluids
+nude
+tongue
+simple_background
+looking_at_viewer
+smile
+canis
+male/male
+claws
+teeth
+male/female
+felid
+muscular
+tuft
+wolf
+dragon
+pawpads
+muscular_male
+legwear
+fox
+grey_body
+precum
+solo_focus
+black_nose
+4_toes
+grey_fur
+shirt
+muscular_anthro
+abs
+wings
+eyelashes
+jewelry
+brown_hair
+raised_tail
+felis
+ambiguous_gender
+submissive
+headgear
+partially_clothed
+white_background
+thigh_highs
+glasses
+seductive
+two_tone_fur
+smaller_male
+equine_penis
+bdsm
+generation_1_pokemon
+larger_male
+chest_tuft
+digitigrade
+fluffy
+headwear
+yellow_fur
+drooling
+monochrome
+pink_nipples
+bulge
+biceps
+bedroom_eyes
+blue_hair
+submissive_male
+narrowed_eyes
+underwear
+girly
+detailed_background
+shirt
+bound