Sense-X / Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training
MIT License
950 stars 100 forks source link

🐛 Bugfix: Add pip [fairscale, scipy, yapf] #123

Closed Hitbee-dev closed 4 months ago

Hitbee-dev commented 5 months ago

ISSUE

When building the Dockerfile from the repository and proceeding with training according to the "Train Co-Deformable-DETR + ResNet-50" Guide, an error occurs stating that three Python packages are not installed.

Below are the logs of the errors that occurred.

1. fairscale not pip install

ERROR MSG:

    import fairscale
ModuleNotFoundError: No module named 'fairscale'

2. scipy not pip install

ERROR MSG:

    raise ImportError('Please run "pip install scipy" '
ImportError: Please run "pip install scipy" to install scipy first.

3. yapf not pip install

ERROR MSG:

    text, _ = FormatCode(text, style_config=yapf_style, verify=True)
TypeError: FormatCode() got an unexpected keyword argument 'verify'

Therefore, I have added the necessary Python packages to the Dockerfile. Now, you can proceed with the training immediately after building the Dockerfile.