ComfyUI Node: load torchvision model

Authored by 807502278

Created

Updated

11 stars

Category

WJNode/Detection

Inputs

model_name
  • resnet18
  • resnet34
  • resnet50
  • resnet101
  • resnet152
  • resnext50_32x4d
  • resnext101_32x8d
  • resnext101_64x4d
  • wide_resnet50_2
  • wide_resnet101_2
  • densenet121
  • densenet169
  • densenet201
  • densenet161
  • efficientnet_b0
  • efficientnet_b1
  • efficientnet_b2
  • mobilenet_v2
  • mobilenet_v3_small
  • mobilenet_v3_large
  • vgg11
  • vgg13
  • vgg16
  • vgg19
download
  • LocalOnly
  • Download
device
  • default
  • cpu
  • cuda:0

Outputs

Similarity

Extension: ComfyUI-WJNodes

Ready to use upon download. No need to install dependencies for the time being. If there are new functions or suggestions, please provide feedback. Attention! The delfile node is not recommended for use on servers. I am not responsible for any losses incurred.

Authored by 807502278

View Nodes

Run ComfyUI workflows in the Cloud!

No downloads or installs are required. Pay only for active GPU usage, not idle time. No complex setups and dependency issues

Learn more