Share ckpt.bin

[kyelok]

Would you be able to share your ckpt.bin pre-trained weights? ckpt.t7 from https://drive.google.com/drive/folders/1xhG0kRH1EX5B9_Iz8gQJb7UNnn_riXi6 doesn't seem to be the same format as yours.

[shodow]

+1

[qzp-user]

+1

[shodow]

according to #6 and #5 , i use this code to convert ckpt.t7 to ckpt.bin

import os
import torch
import numpy as np

model_path="./ckpt.t7"
state_dict = torch.load(model_path)['net_dict']
with open("ckpt.bin", "wb") as f:
    for key in state_dict.keys():
        f.write(state_dict[key].cpu().numpy())

[kumarvis]

Is it possible to get your feature extractor network training code in case we want to use custome object.