keras-team / keras-io

Keras documentation, hosted live at keras.io
Apache License 2.0
2.76k stars 2.04k forks source link

Pre-trained ResNet50 as backbone of YOLOv8 #1545

Closed SaraDadjouy closed 11 months ago

SaraDadjouy commented 1 year ago

Hello! I want to do an object detection task and read this Efficient Object Detection with YOLOV8 and KerasCV.

I have a pre-trained ResNet50 model and want to use it as the backbone with YOLOv8Detector. But I'm not sure if it is even possible. Would you please help me with this on how can I do this?

sachinprasadhs commented 1 year ago

I'm not sure if you can use ResNet50 as backbone for YOLOv8Detector.

Probably you can give it a try by providing ResNet backbone in the below code snippet.

 yolo = keras_cv.models.YOLOV8Detector(
    num_classes=len(class_mapping),
    bounding_box_format="xyxy",
    backbone=backbone,
    fpn_depth=1,
)

Refer the YOLOv8 architecture here https://blog.roboflow.com/whats-new-in-yolov8/

github-actions[bot] commented 11 months ago

This issue is stale because it has been open for 14 days with no activity. It will be closed if no further activity occurs. Thank you.

github-actions[bot] commented 11 months ago

This issue was closed because it has been inactive for 28 days. Please reopen if you'd like to work on this further.