This commit is contained in:
琉璃月光
2025-10-21 14:11:52 +08:00
parent 349449f2b7
commit df7c0730f5
363 changed files with 5386 additions and 578 deletions

View File

@ -91,8 +91,8 @@ def process_obb_images(model_path, image_dir, output_dir="./inference_results",
# ------------------- 测试调用 -------------------
if __name__ == "__main__":
MODEL_PATH = r'best.pt'
IMAGE_SOURCE_DIR = r"./test_image"
MODEL_PATH = r'obb.pt'
IMAGE_SOURCE_DIR = r"/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/obb4/val"
OUTPUT_DIR = "./inference_results"
results = process_obb_images(MODEL_PATH, IMAGE_SOURCE_DIR, OUTPUT_DIR)