bushu
This commit is contained in:
@ -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)
|
||||
|
||||
Reference in New Issue
Block a user