├── LICENSE ├── README.md ├── eval.ipynb ├── mAP ├── main.py ├── output │ ├── classes │ │ ├── mask.png │ │ └── nomask.png │ ├── detection-results-info.png │ ├── ground-truth-info.png │ ├── lamr.png │ ├── mAP.png │ └── output.txt └── scripts │ └── extra │ ├── README.md │ ├── class_list.txt │ ├── convert_dr_darkflow_json.py │ ├── convert_dr_yolo.py │ ├── convert_gt_xml.py │ ├── convert_gt_yolo.py │ ├── convert_keras-yolo3.py │ ├── find_class.py │ ├── intersect-gt-and-dr.py │ └── result.txt ├── make_annotations.ipynb ├── model_data ├── mask_classes.txt ├── mask_train.txt ├── mask_val.txt ├── simhei.ttf └── yolo_anchors.txt ├── nets ├── CSPdarknet.py ├── yolo4.py └── yolo_training.py ├── predict.ipynb ├── testImage_result.png ├── total_loss.png ├── train.ipynb ├── utils └── utils.py └── yolo.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /eval.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/eval.ipynb -------------------------------------------------------------------------------- /mAP/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/main.py -------------------------------------------------------------------------------- /mAP/output/classes/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/output/classes/mask.png -------------------------------------------------------------------------------- /mAP/output/classes/nomask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/output/classes/nomask.png -------------------------------------------------------------------------------- /mAP/output/detection-results-info.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/output/detection-results-info.png -------------------------------------------------------------------------------- /mAP/output/ground-truth-info.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/output/ground-truth-info.png -------------------------------------------------------------------------------- /mAP/output/lamr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/output/lamr.png -------------------------------------------------------------------------------- /mAP/output/mAP.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/output/mAP.png -------------------------------------------------------------------------------- /mAP/output/output.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/output/output.txt -------------------------------------------------------------------------------- /mAP/scripts/extra/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/scripts/extra/README.md -------------------------------------------------------------------------------- /mAP/scripts/extra/class_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/scripts/extra/class_list.txt -------------------------------------------------------------------------------- /mAP/scripts/extra/convert_dr_darkflow_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/scripts/extra/convert_dr_darkflow_json.py -------------------------------------------------------------------------------- /mAP/scripts/extra/convert_dr_yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/scripts/extra/convert_dr_yolo.py -------------------------------------------------------------------------------- /mAP/scripts/extra/convert_gt_xml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/scripts/extra/convert_gt_xml.py -------------------------------------------------------------------------------- /mAP/scripts/extra/convert_gt_yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/scripts/extra/convert_gt_yolo.py -------------------------------------------------------------------------------- /mAP/scripts/extra/convert_keras-yolo3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/scripts/extra/convert_keras-yolo3.py -------------------------------------------------------------------------------- /mAP/scripts/extra/find_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/scripts/extra/find_class.py -------------------------------------------------------------------------------- /mAP/scripts/extra/intersect-gt-and-dr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/scripts/extra/intersect-gt-and-dr.py -------------------------------------------------------------------------------- /mAP/scripts/extra/result.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/mAP/scripts/extra/result.txt -------------------------------------------------------------------------------- /make_annotations.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/make_annotations.ipynb -------------------------------------------------------------------------------- /model_data/mask_classes.txt: -------------------------------------------------------------------------------- 1 | mask 2 | nomask -------------------------------------------------------------------------------- /model_data/mask_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/model_data/mask_train.txt -------------------------------------------------------------------------------- /model_data/mask_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/model_data/mask_val.txt -------------------------------------------------------------------------------- /model_data/simhei.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/model_data/simhei.ttf -------------------------------------------------------------------------------- /model_data/yolo_anchors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/model_data/yolo_anchors.txt -------------------------------------------------------------------------------- /nets/CSPdarknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/nets/CSPdarknet.py -------------------------------------------------------------------------------- /nets/yolo4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/nets/yolo4.py -------------------------------------------------------------------------------- /nets/yolo_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/nets/yolo_training.py -------------------------------------------------------------------------------- /predict.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/predict.ipynb -------------------------------------------------------------------------------- /testImage_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/testImage_result.png -------------------------------------------------------------------------------- /total_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/total_loss.png -------------------------------------------------------------------------------- /train.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/train.ipynb -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/utils/utils.py -------------------------------------------------------------------------------- /yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Bil369/MaskDetect-YOLOv4-PyTorch/HEAD/yolo.py --------------------------------------------------------------------------------