akanazawa / hmr

Project page for End-to-end Recovery of Human Shape and Pose
Other
1.54k stars 395 forks source link

can we retrain model to accept images with resolution higher than 224x224? #129

Closed Naveennavik closed 4 years ago

Naveennavik commented 4 years ago

Hi @akanazawa @AdityaAS , can the model be retrained or fine tuned to accept images with resolution higher than 224x224?

Thanks in advace.

akanazawa commented 4 years ago

Sure! just re-write the data loader to return a bigger image and retrain.

Naveennavik commented 4 years ago

I dont seem image to be resized in dataloader if it so can u say me on what line to change? @akanazawa

akanazawa commented 4 years ago

https://github.com/akanazawa/hmr/blob/bce0ef9b90bd36871d2aff8688b2682170cd365a/src/data_loader.py#L56

Best,

A