Can't get attribute 'DetectionModel' on <module 'models.yolo' from

Summary

Share a clear and concise description of the issue. Aim for 2-3 sentences.

Steps to reproduce

Code snippet:

add code here

If applicable, please provide the steps we should take to reproduce the error or specified behavior.

Expected behavior:

Explain what you expect to happen when you run the code above.

Actual behavior:

Explain the undesired behavior or error you see when you run the code above.
If you’re seeing an error message, share the full contents of the error message here.

Debug info

Requirements file

Using

streamlit

base ----------------------------------------

matplotlib>=3.2.2
numpy>=1.18.5
opencv-python-headless>=4.1.2
Pillow
PyYAML>=5.3.1
scipy>=1.4.1
torch==1.7.1
torchvision>=0.8.1
tqdm>=4.41.0
click==8

logging -------------------------------------

tensorboard>=2.4.1

wandb

plotting ------------------------------------

seaborn>=0.11.0
pandas

export --------------------------------------

coremltools>=4.1

onnx>=1.8.1

scikit-learn==0.19.2 # for coreml quantization

extras --------------------------------------

thop # FLOPS computation
pycocotools>=2.0 # COCO mAP

Links

Additional information

If needed, add any other context about the problem here.