- add input size parameter for inference (169)
Example usages:
python
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
)