Skip to content

Instantly share code, notes, and snippets.

@david-littlefield

david-littlefield/dsfd_error_09_widerface_val.py

Last active November 6, 2020 20:05
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save david-littlefield/4bee33217a80aeef9c8d9eb9f56134df to your computer and use it in GitHub Desktop.
Save david-littlefield/4bee33217a80aeef9c8d9eb9f56134df to your computer and use it in GitHub Desktop.
Download ZIP
Raw
dsfd_error_09_widerface_val.py
# Line 231
# load net
cfg = widerface_640
num_classes = len(WIDERFace_CLASSES) + 1 # +1 background
net = build_ssd('test', cfg['min_dim'], num_classes) # initialize SSD
net.load_state_dict(torch.load(args.trained_model))
#net.cuda() # <----------
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment