HamadYA / GhostFaceNets

This repository contains the official implementation of GhostFaceNets, State-Of-The-Art lightweight face recognition models.
https://ieeexplore.ieee.org/document/10098610
MIT License
194 stars 38 forks source link

Train Custom DataSet #37

Closed Rhee-site closed 11 months ago

Rhee-site commented 11 months ago

First of all, thank you for the good research. I'm trying to train on a custom dataset. The dataset consists of 224x224, and the input_shape has been changed in GhostFaceNets.py. The train code refers to training_scripts.ipynb. I'm using GhostFaceNetV2.

Please provide feedback on whether there are any additional changes that need to be made.

Confirmed errors are as follows: ValueError: input 0 of layer "model" is incompatible with layer: expected shape=(none, 224, 224, 3), found shape=(128, 112, 112, 3)

Am I entering something incorrectly?