Classification algorythm

Hello all !
I am looking for a github code of image classification which take large images (1500x2000 pixels) images at input of neural network (all codes I have found take maximum 224x224 pixels images) and which take account of very imbalanced classes. I must can input my own dataset for training.
Thank you
Best regards