Ayush ChaurasiaandGlenn Jocher
ca290dca24
* Add wandb metric logging and bounding box debugging * Improve formatting, readability * Remove mutliple path for init, improve formatting * Add wandb params * Remove typecasting in bbox coordinates and reformat * Cleanup * add wandb to requirements.txt * minor updates to test.py * general reorg * reduce --log-imgs to 10 * clean wandb import * reverse wandb import assert * add except AssertionError to try import * move wandb init to all global ranks * replace print() with logger.info() * replace print() with logger.info() * move wandb.init() bug fix * project PosixPath to basename bug fix Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
31 lines
624 B
Plaintext
Executable File
31 lines
624 B
Plaintext
Executable File
# pip install -r requirements.txt
|
|
|
|
# base ----------------------------------------
|
|
Cython
|
|
matplotlib>=3.2.2
|
|
numpy>=1.18.5
|
|
opencv-python>=4.1.2
|
|
pillow
|
|
PyYAML>=5.3
|
|
scipy>=1.4.1
|
|
tensorboard>=2.2
|
|
torch>=1.6.0
|
|
torchvision>=0.7.0
|
|
tqdm>=4.41.0
|
|
|
|
# logging -------------------------------------
|
|
# wandb
|
|
|
|
# coco ----------------------------------------
|
|
# pycocotools>=2.0
|
|
|
|
# export --------------------------------------
|
|
# packaging # for coremltools
|
|
# coremltools==4.0
|
|
# onnx>=1.7.0
|
|
# scikit-learn==0.19.2 # for coreml quantization
|
|
|
|
# extras --------------------------------------
|
|
# thop # FLOPS computation
|
|
# seaborn # plotting
|