search

tianyu0207 / IGD

Official code for 'Deep One-Class Classification via Interpolated Gaussian Descriptor' [AAAI 2022 Oral]
63 stars 8 forks source link

Model Prediction #3

Open BoltonAthitDavies opened 2 years ago

BoltonAthitDavies commented 2 years ago

How to make a model's prediction on new images? Thanks

tianyu0207 commented 2 years ago

How to make a model's prediction on new images? Thanks

you can just change the dataloader and train based on your own data. After that you can simply inference based on your teained model.

BoltonAthitDavies commented 2 years ago

So my problem is I cannot find model variable for using model.load_state_dict. How should I solve it ?

tianyu0207 commented 2 years ago

So my problem is I cannot find model variable for using model.load_state_dict. How should I solve it ?

Hi we have just updated the checkpoint on the Github https://github.com/tianyu0207/IGD/blob/main/Encoder_KD_ckpt

BoltonAthitDavies commented 2 years ago

Hi, Sorry for a bunch of questions. For my understanding, Training with p256 will save the model in Encoder_KD_ckpt, Am I right?

Thank tou

ในวันที่ พฤ. 9 มิ.ย. 2022 เวลา 13:58 Yu Tian @.***> เขียนว่า:

So my problem is I cannot find model variable for using model.load_state_dict. How should I solve it ?

Hi we have just updated the checkpoint on the Github https://github.com/tianyu0207/IGD/blob/main/Encoder_KD_ckpt

— Reply to this email directly, view it on GitHub https://github.com/tianyu0207/IGD/issues/3#issuecomment-1150744201, or unsubscribe https://github.com/notifications/unsubscribe-auth/ARJ3SNZNVCVDHZUILAHLIT3VOGIXTANCNFSM5XJ6D7WQ . You are receiving this because you authored the thread.Message ID: @.***>

xiabc612 commented 3 months ago

Is there an inference code for new test images? Thanks