Complex-YOLOv3 icon indicating copy to clipboard operation
Complex-YOLOv3 copied to clipboard

Module doesn't work if BEV RGB images sizes are not square?

Open chowkamlee81 opened this issue 5 years ago • 5 comments

I need to change FOV from [0,50] and [-25,25] to [0,40] and [-40,40]. In trhis case iam assuming BEV RGB images are of size 480*960, then YOLO modules crashes. KIndly let us know whether YOLOv3 works if images are not of shape square.

chowkamlee81 avatar Dec 02 '19 05:12 chowkamlee81

@ghimiredhikura , can you please suggest

chowkamlee81 avatar Dec 04 '19 04:12 chowkamlee81

@chowkamlee81 did u re-train the model after changing the FOV ?

abhigoku10 avatar Dec 04 '19 06:12 abhigoku10

@abhigoku10

I tried to retrain by changing the FOV but module got crashed since Anchor gride size is of squared size. Is there is any possibility it will work if we change our resolution to not to a squared size?

chowkamlee81 avatar Dec 04 '19 10:12 chowkamlee81

@chowkamlee81 ideally the complex yolo takes in non squared resolution, i think we need to generate new anchor boxes for your custom dataset . Can you share the error which your getting

abhigoku10 avatar Dec 05 '19 03:12 abhigoku10

Hi @abhigoku10, @chowkamlee81,

Yes currently the model only supports square size, as size is passed as single value (not width and height separately) and used as same value across x and y axis. If you want to make it work with non square size you have to pass height and width separately in models.py file and change the code accordingly.

PS: There was small error regarding input image size in train.py, fixed with last commit.

Best, Deepak

ghimiredhikura avatar Dec 06 '19 01:12 ghimiredhikura