y / weights.py
Tushdm's picture
ss
94c42be
Raw
History Blame Contribute Delete
453 Bytes
import requests
import os.path
def weights():
path = 'models/yolov7.pt'
check_file = os.path.isfile(path)
if check_file == True:
pass
else:
URL = "https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt"
response = requests.get(URL)
open(path, "wb").write(response.content)
if __name__ == "__main__":
try:
weights()
except Exception as e:
print(f'error is {e}')