Skip to content

v0.6.2

Compare
Choose a tag to compare
@fcakyon fcakyon released this 13 Jul 15:58
· 368 commits to main since this release
9501cd8
  • add input size parameter for inference (#169)

Example usages:

detection_model = Yolov5DetectionModel(
    model_path=yolov5_model_path,
    device="cpu", # or 'cuda'
)

result = detection_model.perform_inference(
    image,
    image_size=1280
)

result = get_prediction(
    "demo_data/small-vehicles1.jpeg",
    detection_model,
    image_size=1280
)

result = get_sliced_prediction(
    "demo_data/small-vehicles1.jpeg",
    detection_model,
    image_size=1280,
    slice_height = 256,
    slice_width = 256,
    overlap_height_ratio = 0.2,
    overlap_width_ratio = 0.2
)