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
使用自制数据集训练时出错 #2546
Comments
在群里问道 如果使用飞桨 要把 分辨率统一调成112x112;;网络结构里都写死了,最后一个卷积是7x7 |
Add transforms.Resize((112,112)) into dataset transform |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
我想用 insightface 做一个 牛脸的认证, blazeface 训练成功了;但是;;;第二步;;;/home/aistudio/insightface/recognition/arcface_paddle
!python tools/train.py
--config_file configs/ms1mv2_mobileface.py
--embedding_size 128
--sample_ratio 1.0
--loss ArcFace
--batch_size 5
--dataset MS1M_v2
--num_classes 54
--data_dir MS1M_v2/
--label_file MS1M_v2/label.txt
--fp16 False;;;报 File "/home/aistudio/insightface/recognition/arcface_paddle/dynamic/train.py", line 200, in train
for step, (img, label) in enumerate(train_loader): raise ValueError('all input arrays must have the same shape')
ValueError: all input arrays must have the same shape;;我已经把图片缩放为统一大小为啥还是会出现这个错;;;和embedding_size有关系吗
The text was updated successfully, but these errors were encountered: