pokouqiege/models/line_detect/predict_demo.py

from models.line_detect.line_net import linenet_resnet18_fpn

if __name__ == '__main__':
    model=linenet_resnet18_fpn()
    model.load_best_model(r'E:\projects\tmp\MultiVisionModels\models\line_detect\train_results\20250515_173829\weights\best_val.pth')

    img_path=r"\\192.168.50.222\share\rlq\datasets\修订513pcd转换彩图标注后汇总\2025-05-13-08-37-48_LaserData_ID019504_color.jpg"

    model.predict(img_path)