├── .gitignore ├── dataset ├── coco.py └── config.py ├── demo_yolact_yolov5.py ├── demo_yolov5.py ├── detect.py ├── eval.py ├── log.md ├── models ├── __init__.py ├── backbone.py ├── build_yolact.py ├── common.py ├── experimental.py ├── multi_loss.py ├── yolo.py ├── yolov5_mask.py └── yolov5m.yaml ├── readme.md ├── train.py └── utils ├── __init__.py ├── activations.py ├── augmentations.py ├── box_utils.py ├── cython_nms.pyx ├── datasets.py ├── datasets_robust.py ├── functions.py ├── general.py ├── google_utils.py ├── infer_utils.py ├── labelme2coco.py ├── output_utils.py ├── pascal2coco.py ├── timer.py ├── tmp.py └── torch_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/.gitignore -------------------------------------------------------------------------------- /dataset/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/dataset/coco.py -------------------------------------------------------------------------------- /dataset/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/dataset/config.py -------------------------------------------------------------------------------- /demo_yolact_yolov5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/demo_yolact_yolov5.py -------------------------------------------------------------------------------- /demo_yolov5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/demo_yolov5.py -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/detect.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/eval.py -------------------------------------------------------------------------------- /log.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/log.md -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/models/backbone.py -------------------------------------------------------------------------------- /models/build_yolact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/models/build_yolact.py -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/models/common.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/multi_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/models/multi_loss.py -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/models/yolo.py -------------------------------------------------------------------------------- /models/yolov5_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/models/yolov5_mask.py -------------------------------------------------------------------------------- /models/yolov5m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/models/yolov5m.yaml -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/readme.md -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/augmentations.py -------------------------------------------------------------------------------- /utils/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/box_utils.py -------------------------------------------------------------------------------- /utils/cython_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/cython_nms.pyx -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/datasets_robust.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/datasets_robust.py -------------------------------------------------------------------------------- /utils/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/functions.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/google_utils.py -------------------------------------------------------------------------------- /utils/infer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/infer_utils.py -------------------------------------------------------------------------------- /utils/labelme2coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/labelme2coco.py -------------------------------------------------------------------------------- /utils/output_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/output_utils.py -------------------------------------------------------------------------------- /utils/pascal2coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/pascal2coco.py -------------------------------------------------------------------------------- /utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/timer.py -------------------------------------------------------------------------------- /utils/tmp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/tmp.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/yolov5_mask/HEAD/utils/torch_utils.py --------------------------------------------------------------------------------