You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
DroneDetector/train_scripts/requirements-train.txt

25 lines
666 B
Plaintext

# Base deps for notebooks in train_scripts/.
#
# Install order for this repo:
# python -m pip install -r deploy/requirements/nn_gpu_pinned.txt
# python -m pip install -r train_scripts/requirements-train.txt
# python -m pip install -e ./torchsig --no-deps
#
# Notes:
# - Do not install `torchsig` from PyPI: the project uses the local repo copy.
# - Main NN service pins come from deploy/requirements/nn_common.txt.
# - For notebooks use GUI OpenCV, so this file uses `opencv-python`, not headless.
ipykernel
jupyterlab
numpy==2.1.3
matplotlib==3.10.0
tqdm==4.67.1
PyYAML==6.0.2
mlconfig==0.3.2
scikit-learn==1.6.0
opencv-python==4.10.0.84
pandas
Pillow
scipy