Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

RetinaNet with custom backbone and custom dataset #773

Closed
adityaroy10 opened this issue Apr 11, 2024 · 3 comments
Closed

RetinaNet with custom backbone and custom dataset #773

adityaroy10 opened this issue Apr 11, 2024 · 3 comments

Comments

@adityaroy10
Copy link

adityaroy10 commented Apr 11, 2024

How to form the dataset for training?

for :
model.fit( X, Y) #where model is the retinanet object with appropriate backbone using keras_cv.models.RetinaNet()

X would be the array of images.
So for X0 (first image), will Y0 be array of [class, xmin, xmax, ymin, ymax] for all objects in the image?
What is the correct format?

Right now, I have the annotations saved in [class, xmin, xmax, ymin, ymax] format in a separate text file for each image

Is there any alternative for such case?

Thank you for any help.

@tilakrayal
Copy link
Collaborator

@adityaroy10,
RetinaNet, a popular single-stage detector, which is accurate and runs fast. RetinaNet uses a feature pyramid network to efficiently detect objects at multiple scales and introduces a new loss, the Focal loss function, to alleviate the problem of the extreme foreground-background class imbalance. Also there is COCO2017 dataset which has around 118k images for the training and testing.

https://keras.io/examples/vision/retinanet/
Also there is the data structure of the coco-dataset for the own custom data using TFDS.
https://www.tensorflow.org/datasets/add_dataset#version

Looks like this issue is not related to tf-keras. Could you please try to file an issue on tensorflow forum for such type of issues. Thank you!

Copy link

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 github-actions bot added the stale label Apr 30, 2024
Copy link

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants