Detection object with custom YOLOv5 model by using SAHI

[PawelKinczyk]

I try to use SAHI library for object detection with my custom trained YOLOv5s6 model. I though SAHI support YOLOv5 models but when i try to build detection model i get an error:

Traceback (most recent call last):
  File "C:\Users\pawel\Documents\GitHub\AECVision\aec-env\lib\site-packages\sahi\models\yolov5.py", line 29, in load_model
    model = yolov5.load(self.model_path, device=self.device)
AttributeError: module 'yolov5' has no attribute 'load'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "c:\Users\pawel\Documents\GitHub\AECVision\wall_detection_export_with_sahi.py", line 84, in <module>
    detection_model = AutoDetectionModel.from_pretrained(
  File "C:\Users\pawel\Documents\GitHub\AECVision\aec-env\lib\site-packages\sahi\auto_model.py", line 66, in from_pretrained
    return DetectionModel(
  File "C:\Users\pawel\Documents\GitHub\AECVision\aec-env\lib\site-packages\sahi\models\base.py", line 67, in __init__
    self.load_model()
  File "C:\Users\pawel\Documents\GitHub\AECVision\aec-env\lib\site-packages\sahi\models\yolov5.py", line 32, in load_model
    raise TypeError("model_path is not a valid yolov5 model path: ", e)
TypeError: ('model_path is not a valid yolov5 model path: ', AttributeError("module 'yolov5' has no attribute 'load'"))

I have my custom model weight in 'path_model'

Below is my code:

# Upload pdf and change to jpg
path_pdf = Path("wall_detection_export/upload_pdf")
path_convert_pdf = Path("wall_detection_export/convert_pdf")
path_export_txt = Path("wall_detection_export/export_txt")
path_model = Path("train_results/model_12classes/weights/best.pt")

converter = Convert_pdf(path_pdf=path_pdf)
convert_file = converter.save_image(path_convert_pdf)

# Set detection model
detection_model = AutoDetectionModel.from_pretrained(
    model_type='yolov5',
    model_path=path_model,
    confidence_threshold=0.3,
    device="cuda", # or 'cuda:0'
)

# Slice prediction with sahi
result = get_sliced_prediction(
    convert_file,
    detection_model,
    slice_height = 1280,
    slice_width = 1280,
    overlap_height_ratio = 0.2,
    overlap_width_ratio = 0.2
)
result.export_visuals(export_dir=path_export_txt) 

How can i fix this issue? Thanks for help!

[phhh-xh]

Has your custom model modified the network structure?

[6xw]

Does SAHI support the structurally modified Yolov5?

[fcakyon]

@6xw no it does not support