Image Classification
Transformers
Safetensors
PyTorch
English
Chinese
beit
ai-detection
ai-image-detection
deepfake-detection
fake-image-detection
ai-art-detection
stable-diffusion-detection
midjourney-detection
dall-e-detection
image-forensics
digital-art-verification
vit
computer-vision
Eval Results (legacy)
Update README.md
Browse files
README.md
CHANGED
|
@@ -164,7 +164,6 @@ def detect_image(image_path):
|
|
| 164 |
outputs = model(**inputs)
|
| 165 |
probs = torch.softmax(outputs.logits, dim=-1)[0]
|
| 166 |
|
| 167 |
-
# Top-1 决定 + 置信度
|
| 168 |
top_idx = probs.argmax().item()
|
| 169 |
top_source = model.config.id2label[str(top_idx)]
|
| 170 |
top_confidence = probs[top_idx].item()
|
|
|
|
| 164 |
outputs = model(**inputs)
|
| 165 |
probs = torch.softmax(outputs.logits, dim=-1)[0]
|
| 166 |
|
|
|
|
| 167 |
top_idx = probs.argmax().item()
|
| 168 |
top_source = model.config.id2label[str(top_idx)]
|
| 169 |
top_confidence = probs[top_idx].item()
|