Personal Protective Equipment Dataset (PPED)

<strong>Personal Protective Equipment Dataset (PPED)</strong> This dataset serves as a benchmark for PPE in chemical plants<br> We provide datasets and experimental results. <strong>1. The dataset</strong> We produced a data set based on the actual needs and relevant regulations in chemical plants.<br> The standard GB 39800.1-2020 formulated by the Ministry of Emergency Management of the People’s Republic of China defines the protective requirements for plants and chemical laboratories.<br> The complete dataset is contained in the folder <code>PPED/data</code>. <strong>1.1. Image collection</strong> We took more than 3300 pictures.<br> We set the following different characteristics, including different environments, different distances, different lighting conditions, different angles, and the diversity of the number of people photographed. Backgrounds: There are 4 backgrounds, including office, near machines, factory and regular outdoor scenes. Scale: By taking pictures from different distances, the captured PPEs are classified in small, medium and large scales. Light: Good lighting conditions and poor lighting conditions were studied. Diversity: Some images contain a single person, and some contain multiple people. Angle: The pictures we took can be divided into front and side. A total of more than 3300 photos were taken in the raw data under all conditions.<br> All images are located in the folder “PPED/data/JPEGImages”. <strong>1.2. Label</strong> We use Labelimg as the labeling tool, and we use the PASCAL-VOC labelimg format.<br> Yolo use the txt format, we can use <code>trans_voc2yolo.py</code> to convert the XML file in PASCAL-VOC format to txt file.<br> Annotations are stored in the folder <code>PPED/data/Annotations</code> <strong>1.3. Dataset Features</strong> The pictures are made by us according to the different conditions mentioned above.<br> The file <code>PPED/data/feature.csv</code> is a CSV file which notes all the .os of all the image. It records every feature of the picture, including lighting conditions, angles, backgrounds, number of people and scale. 1.4. Dataset Division The data set is divided into 9:1 training set and test set. <strong>2. Baseline Experiments</strong> We provide baseline results with five models, namely Faster R-CNN ®, Faster R-CNN (M), SSD, YOLOv3-spp, and YOLOv5.<br> All code and results is given in folder <code>PPED/experiment</code>. <strong>2.1. Environment and Configuration:</strong> Intel Core i7-8700 CPU NVIDIA GTX1060 GPU 16 GB of RAM Python: 3.8.10 pytorch: 1.9.0 pycocotools: pycocotools-win Windows 10 <strong>2.2. Applied Models</strong> The source codes and results of the applied models is given in folder <code>PPED/experiment</code> with sub-folders corresponding to the model names. <strong>2.2.1. Faster R-CNN</strong> Faster R-CNN backbone: resnet50+fpn We downloaded the pre-training weights from https://download.pytorch.org/models/fasterrcnn_resnet50_fpn_coco-258fb6c6.pth. We modified the dataset path, training classes and training parameters including batch size. We run <code>train_res50_fpn.py</code> start training. Then, the weights are trained by the training set. Finally, we validate the results on the test set. backbone: mobilenetv2 the same training method as resnet50+fpn, but the effect is not as good as resnet50+fpn, so it is directly discarded. The Faster R-CNN source code used in our experiment is given in folder <code>PPED/experiment/Faster R-CNN</code>.<br> The weights of the fully-trained Faster R-CNN (R), Faster R-CNN (M) model are stored in file <code>PPED/experiment/trained_models/resNetFpn-model-19.pth and </code>mobile-model.pth.<br> The performance measurements of Faster R-CNN (R) Faster R-CNN (M) are stored in folder <code>PPED/experiment/results/Faster RCNN(R)and Faster RCNN(M)</code>. <strong>2.2.2. SSD</strong> backbone: resnet50 We downloaded pre-training weights from https://download.pytorch.org/models/resnet50-19c8e357.pth. The same training method as Faster R-CNN is applied. The SSD source code used in our experiment is given in folder <code>PPED/experiment/ssd</code>.<br> The weights of the fully-trained SSD model are stored in file <code>PPED/experiment/trained_models/SSD_19.pth</code>.<br> The performance measurements of SSD are stored in folder <code>PPED/experiment/results/SSD</code>. <strong>2.2.3. YOLOv3-spp</strong> backbone: DarkNet53 We modified the type information of the XML file to match our application. We run <code>trans_voc2yolo.py</code> to convert the XML file in VOC format to a txt file. The weights used are: yolov3-spp-ultralytics-608.pt. The YOLOv3-spp source code used in our experiment is given in folder <code>PPED/experiment/YOLOv3-spp</code>.<br> The weights of the fully-trained YOLOv3-spp model are stored in file <code>PPED/experiment/trained_models/YOLOvspp-19.pt</code>.<br> The performance measurements of YOLOv3-spp are stored in folder <code>PPED/experiment/results/YOLOv3-spp</code>. <strong>2.2.4. YOLOv5</strong> backbone: CSP_DarkNet We modified the type information of the XML file to match our application. We run <code>trans_voc2yolo.py</code> to convert the XML file in VOC format to a txt file. The weights used are: yolov5s. The YOLOv5 source code used in our experiment is given in folder <code>PPED/experiment/yolov5</code>.<br> The weights of the fully-trained YOLOv5 model are stored in file <code>PPED/experiment/trained_models/YOLOv5.pt</code>.<br> The performance measurements of YOLOv5 are stored in folder <code>PPED/experiment/results/YOLOv5</code>. 2.3. Evaluation The computed evaluation metrics as well as the code needed to compute them from our dataset are provided in the folder <code>PPED/experiment/eval</code>. <strong>3. Code Sources</strong> Faster R-CNN (R and M) https://github.com/WZMIAOMIAO/deep-learning-for-image-processing/tree/master/pytorch_object_detection/faster_rcnn official code: https://github.com/pytorch/vision/blob/main/torchvision/models/detection/faster_rcnn.py SSD https://github.com/WZMIAOMIAO/deep-learning-for-image-processing/tree/master/pytorch_object_detection/ssd official code: https://github.com/pytorch/vision/blob/main/torchvision/models/detection/ssd.py YOLOv3-spp https://github.com/WZMIAOMIAO/deep-learning-for-image-processing/tree/master/pytorch_object_detection/yolov3-spp YOLOv5 https://github.com/ultralytics/yolov5