草庐IT

SegmentationModel

全部标签

YOLOv5图像分割--SegmentationModel类代码详解

目录​编辑SegmentationModel类DetectionModel类推理阶段DetectionModel--forward()BaseModel--forward() Segment类Detect--forward  SegmentationModel类定义model将会调用models/yolo.py中的类SegmentationModel。该类是继承父类--DetectionModel类。classSegmentationModel(DetectionModel):#SegmentationModel这个类是继承了DetectionModel这个类#YOLOv5segmentati