Remove /weights directory (#3659)
* Remove `/weights` directory * cleanup
This commit is contained in:
1 parent
9b6dba6207
commit
2754adad46
4 files changed
+7
-10
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user