训练模型:train_yolov8.py和train_faster_rcnn.py分别负责加载数据、初始化模型、损失函数和优化器,并进行训练。 推理和可视化:infer_yolov8.py和infer_faster_rcnn.py用于加载训练好的模型并对新图像进行预测,显示预测结果。 评价指标:evaluate_yolov8.py和evaluate_faster_rcnn.py计算模型的评价指标,如IoU和mA...
使用自制的碎玻璃数据集在 Faster R-CNN 算法上进行了训练，Faster R-CNN 算法所依赖的库是 OpenCV-...
我拿faster-RCNN-resnet50为例,首先需要修改$mytrain中配置文件faster_rcnn_resnet50_coco.config,需要修改六个地方: num_classes:修改为自己的类别数 将所有PATH_TO_BECONFIGURED的地方修改为自己之前设置的路径(fine_tune_checkpoint路径为之前下载的预训练模型所在路径;两个input_path路径设置为models/objectdetecti...
训练模型:train_yolov8.py和train_faster_rcnn.py分别负责加载数据、初始化模型、损失函数和优化器,并进行训练。 推理和可视化:infer_yolov8.py和infer_faster_rcnn.py用于加载训练好的模型并对新图像进行预测,显示预测结果。 评价指标:evaluate_yolov8.py和evaluate_faster_rcnn.py计算模型的评价指标,如IoU和mA...