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

使用自制数据集训练时出错 #2546

Open
enjoyloveless opened this issue Mar 26, 2024 · 2 comments
Open

使用自制数据集训练时出错 #2546

enjoyloveless opened this issue Mar 26, 2024 · 2 comments

Comments

@enjoyloveless
Copy link

我想用 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有关系吗

@enjoyloveless
Copy link
Author

在群里问道 如果使用飞桨 要把 分辨率统一调成112x112;;网络结构里都写死了,最后一个卷积是7x7

@airkid
Copy link

airkid commented Apr 15, 2024

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
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants