Structure

hardhat_detection_model/
      ├── README.md
      ├── requirements.txt
      ├── data_gathering/
      │          ├── base_data.py
      │          └── inference_data.py
      ├── training/
      │          ├── train_test_split.py
      │          └── train.py
      ├── inference/
      │          └── inference.py
      ├── model_store/
      └── docker_deployment/