pull/3730/head
fcakyon 2021-06-22 14:33:38 +03:00 committed by GitHub
parent b83e1a4adc
commit 0e2d0d54d7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -43,7 +43,7 @@ def run(weights='./yolov5s.pt', # weights path
# Load PyTorch model
device = select_device(device)
assert not (device.type == 'cpu' and opt.half), '--half only compatible with GPU export, i.e. use --device 0'
assert not (device.type == 'cpu' and half), '--half only compatible with GPU export, i.e. use --device 0'
model = attempt_load(weights, map_location=device) # load FP32 model
labels = model.names