mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-27 17:14:22 +08:00
Fix code scanning alert - Incomplete URL substring sanitization (#3054)
This commit is contained in:
parent
31281400dd
commit
8ac8ff72ae
@ -119,7 +119,7 @@ class YOLO:
|
||||
def is_hub_model(model):
|
||||
"""Check if the provided model is a HUB model."""
|
||||
return any((
|
||||
model.startswith('https://hub.ultra'), # i.e. https://hub.ultralytics.com/models/MODEL_ID
|
||||
model.startswith('https://hub.ultralytics.com/models/'), # i.e. https://hub.ultralytics.com/models/MODEL_ID
|
||||
[len(x) for x in model.split('_')] == [42, 20], # APIKEY_MODELID
|
||||
len(model) == 20 and not Path(model).exists() and all(x not in model for x in './\\'))) # MODELID
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user