Remove /weights directory (#3659)

* Remove `/weights` directory

* cleanup
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-06-17 14:44:12 +02:00
1 parent 9b6dba6207
commit 2754adad46
4 files changed
+7 -10

No files matched your search

+2 -2
View File
@@ -63,8 +63,8 @@ def detect(weights='yolov5s.pt', # model.pt path(s)
# Second-stage classifier
classify = False
if classify:
modelc = load_classifier(name='resnet101', n=2) # initialize
modelc.load_state_dict(torch.load('weights/resnet101.pt', map_location=device)['model']).to(device).eval()
modelc = load_classifier(name='resnet50', n=2) # initialize
modelc.load_state_dict(torch.load('resnet50.pt', map_location=device)['model']).to(device).eval()
# Set Dataloader
vid_path, vid_writer = None, None