updates
This commit is contained in:
parent
7a2d356297
commit
68b50f5cb6
|
@ -251,7 +251,12 @@ def load_darknet_weights(self, weights, cutoff=-1):
|
|||
# Try to download weights if not available locally
|
||||
if not os.path.isfile(weights):
|
||||
try:
|
||||
os.system('wget https://pjreddie.com/media/files/' + weights_file + ' -O ' + weights)
|
||||
url = 'https://pjreddie.com/media/files/' + weights_file
|
||||
print('Downloading ' + url + ' to ' + weights)
|
||||
os.system('curl ' + url + ' -o ' + weights)
|
||||
import requests
|
||||
r = requests.get(url)
|
||||
|
||||
except IOError:
|
||||
print(weights + ' not found.\nTry https://drive.google.com/drive/folders/1uxgUBemJVw9wZsdpboYbzUN4bcRhsuAI')
|
||||
|
||||
|
|
|
@ -11,8 +11,7 @@ from PIL import Image
|
|||
from tqdm import tqdm
|
||||
from pathlib import Path
|
||||
|
||||
from . import torch_utils
|
||||
from . import google_utils
|
||||
from . import torch_utils # , google_utils
|
||||
|
||||
matplotlib.rc('font', **{'size': 11})
|
||||
|
||||
|
|
Loading…
Reference in New Issue