Quellcode durchsuchen

Add check_python() (#3088)

* Add check_python()

Checks python version against minimum version of 3.7.0.

* remove packaging dependency

* refactor import
modifyDataloader
Glenn Jocher GitHub vor 3 Jahren
Ursprung
Commit
57b0d3a605
Es konnte kein GPG-Schlüssel zu dieser Signatur gefunden werden GPG-Schlüssel-ID: 4AEE18F83AFDEB23
1 geänderte Dateien mit 11 neuen und 1 gelöschten Zeilen
  1. +11
    -1
      utils/general.py

+ 11
- 1
utils/general.py Datei anzeigen

@@ -16,6 +16,7 @@ from pathlib import Path
import cv2
import numpy as np
import pandas as pd
import pkg_resources as pkg
import torch
import torchvision
import yaml
@@ -107,10 +108,19 @@ def check_git_status():
print(e)


def check_python(minimum='3.7.0', required=True):
# Check current python version vs. required python version
current = platform.python_version()
result = pkg.parse_version(current) >= pkg.parse_version(minimum)
if required:
assert result, f'Python {minimum} required by YOLOv5, but Python {current} is currently installed'
return result


def check_requirements(requirements='requirements.txt', exclude=()):
# Check installed dependencies meet requirements (pass *.txt file or list of packages)
import pkg_resources as pkg
prefix = colorstr('red', 'bold', 'requirements:')
check_python() # check python version
if isinstance(requirements, (str, Path)): # requirements.txt file
file = Path(requirements)
if not file.exists():

Laden…
Abbrechen
Speichern