Update google_utils.py (ultralytics#2017)

KMint1819 · Jan 22, 2021 · c7200d4 · c7200d4
1 parent 67104e8
commit c7200d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/google_utils.py b/utils/google_utils.py
@@ -26,8 +26,8 @@ def attempt_download(file, repo='ultralytics/yolov5'):
             assets = [x['name'] for x in response['assets']]  # release assets, i.e. ['yolov5s.pt', 'yolov5m.pt', ...]
             tag = response['tag_name']  # i.e. 'v1.0'
         except:  # fallback plan
-            assets = ['yolov5.pt', 'yolov5.pt', 'yolov5l.pt', 'yolov5x.pt']
-            tag = subprocess.check_output('git tag', shell=True).decode('utf-8').split('\n')[-2]
+            assets = ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']
+            tag = subprocess.check_output('git tag', shell=True).decode().split()[-1]
 
         name = file.name
         if name in assets: