本文共 2843 字，大约阅读时间需要 9 分钟。

????????OpenCV?AI???????????????????????

?????

????????YOLOv5?Mask RCNN???????????

??

?????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????Mask RCNN?Yolov5???????

????

?1??????

????????????????????Beautiful soup?adobe?Istock photo??????????????????????????????????????????Istock photo?Shutter photo?Adobe???????????80?????????350?????????????900???????????

?2???Mask RCNN??????

?????????????????????Mask RCNN???????????????????????????????????????????????Faster RCNN?????Faster RCNN?????????????????????????Mask RCNN??????????????????

Mask RCNN???????????

???

??????

????

ROI??

??Mask RCNN???????????????????????????????????????????????????????????????????

???????Mask RCNN?????????

???????

importing libraries

import pandas as pdimport numpy as npimport cv2import osimport refrom PIL import Imageimport albumentations as Afrom albumentations.pytorch.transforms import ToTensorV2import torchimport torchvisionfrom torchvision.models.detection.faster_rcnn import FastRCNNPredictorfrom torchvision.models.detection import FasterRCNNfrom torchvision.models.detection.rpn import AnchorGeneratorfrom torch.utils.data import DataLoader, Datasetfrom torch.utils.data.sampler import SequentialSamplerfrom matplotlib import pyplot as plt

??????

?????????.csv???????????x?y?w?h??????????????make-sense??????

????img_dir????????????

?????

????????Albumentations???????

?????

?????Resnet50???Mask RCNN?

???????Averager??????????????????????????????

?????

???????cuda????GPU?????????????weight_decay=0.0005?momentum=0.9???????0.05???

???Yolo??????????300?epoch???????????????????50?epoch???????????????????30?epoch??????????

??????????????80???????10???????????

??Mask RCNN????????????????????????????????????????

?3???Yolov5??????

Yolo??????????Ultralytics[github]?????????????????????Yolov5?Yolov4?????????????????????

???????????Yolov5???????????google collab?????????

???????????????????????????

??????????????????Yolo?????????????????????????

????????????????????

???????

?????Yolo?????yaml???????????????????????????????????????????16?????320*320?????

???Yolo??????????300?epoch???????????????????50?epoch???????????????????30?epoch??????????

????????????Yolo????????????10?epoch???300????

???

??????4?????????????????????????????????????

????????????????????Yolo????????????????????????????????(IOU)?

????????Yolov5?????50?epoch?????

????????20?epoch???????????????????????1000????Yolo??????????????????????????

??

??????Yolov5?Mask RCNN???????????????????????????Yolov5??????????????????Yolov5?????????????????????????????????????????????????????????????????????????????????????????????

??????????????????

???????????????????????

?????????????????????????????

THE END!

转载地址：http://eesfk.baihongyu.com/