bemoregtandClaude Sonnet 4.6
a97ca0f3ab
Adds three ComfyUI nodes for adversarial example generation using ResNet18: - LoadResNet18: loads pretrained ImageNet weights - AdversarialAttack: FGSM and PGD untargeted/targeted attacks - ClassifyImage: Top-K inference helper node Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
16 lines
469 B
Python
16 lines
469 B
Python
from .nodes import LoadResNet18, AdversarialAttack, ClassifyImage
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"LoadResNet18": LoadResNet18,
|
|
"AdversarialAttack": AdversarialAttack,
|
|
"ClassifyImage": ClassifyImage,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"LoadResNet18": "Load ResNet18",
|
|
"AdversarialAttack": "Adversarial Attack (FGSM/PGD)",
|
|
"ClassifyImage": "Classify Image (ResNet18)",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|