python evaluate_faster_rcnn.py 八、代码说明 数据配置文件:data/power_line_defect.yaml指定了数据集的路径和类别信息。 训练模型:train_yolov8.py和train_faster_rcnn.py分别负责加载数据、初始化模型、损失函数和优化器,并进行训练。 推理和可视化:infer_yolov8.py和infer_faster_rcnn.py用于加载训练好的模型...
我拿faster-RCNN-resnet50为例,首先需要修改$mytrain中配置文件faster_rcnn_resnet50_coco.config,需要修改六个地方: num_classes:修改为自己的类别数 将所有PATH_TO_BECONFIGURED的地方修改为自己之前设置的路径(fine_tune_checkpoint路径为之前下载的预训练模型所在路径;两个input_path路径设置为models/objectdetecti...
python evaluate_faster_rcnn.py 八、代码说明 数据配置文件:data/power_line_defect.yaml指定了数据集的路径和类别信息。 训练模型:train_yolov8.py和train_faster_rcnn.py分别负责加载数据、初始化模型、损失函数和优化器,并进行训练。 推理和可视化:infer_yolov8.py和infer_faster_rcnn.py用于加载训练好的模型...