cd51e1137b
Multi-GPU update with custom collate function to allow variable size target vector per image without needing to pad targets.
9 lines
189 B
Plaintext
Executable File
9 lines
189 B
Plaintext
Executable File
# pip3 install -U -r requirements.txt
|
|
# conda install numpy opencv matplotlib && conda install pytorch torchvision -c pytorch
|
|
numpy
|
|
opencv-python
|
|
torch >= 1.0.0
|
|
matplotlib
|
|
pycocotools
|
|
tqdm
|