Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

pip_package_creation_second_attempt #3773

Merged
merged 20 commits into from
Jun 26, 2021
Prev Previous commit
more changes 3
  • Loading branch information
SkalskiP committed Jun 26, 2021
commit 7dae15748263ddc8491683305020afd5eb663a8d
3 changes: 1 addition & 2 deletions hubconf.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,7 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
from yolov5.utils.google_utils import attempt_download
from yolov5.utils.torch_utils import select_device

check_requirements(requirements=Path(__file__).parent / 'requirements.txt',
exclude=('tensorboard', 'thop', 'opencv-python'))
check_requirements(exclude=('tensorboard', 'thop', 'opencv-python'))
set_logging(verbose=verbose)

fname = Path(name).with_suffix('.pt') # checkpoint filename
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ pandas
# export --------------------------------------
# coremltools>=4.1
# onnx>=1.9.0
# scikit-learn==0.19.2 # for coreml quantization
# scikit-learn>=0.24 # for coreml quantization

# extras --------------------------------------
# Cython # for pycocotools https://github.com/cocodataset/cocoapi/issues/172
Expand Down
2 changes: 1 addition & 1 deletion yolov5/utils/general.py
Original file line number Diff line number Diff line change
Expand Up @@ -146,7 +146,7 @@ def check_python(minimum='3.6.2', required=True):
return result


def check_requirements(requirements='requirements.txt', exclude=()):
def check_requirements(requirements=Path(__file__).parents[2] / 'requirements.txt', exclude=()):
# Check installed dependencies meet requirements (pass *.txt file or list of packages)
prefix = colorstr('red', 'bold', 'requirements:')
check_python() # check python version
Expand Down