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/