Problems Training SMOKE on custom dataset

lzccccc / SMOKE

SMOKE: Single-Stage Monocular 3D Object Detection via Keypoint Estimation

MIT License

696 stars 177 forks source link

Problems Training SMOKE on custom dataset #53

Closed yqchau closed 3 years ago

yqchau commented 3 years ago

I am training SMOKE on my own dataset with size (1920, 1208) and there are errors in the backbone. Note that I have changed the image size in the config. Please help.

ElectronicElephant commented 3 years ago

IMHO, you should make sure that the input size (w, h) % 32 == 0

yqchau commented 3 years ago

Yes, it's been solved, thanks!