can't find "create_dataset" this file
Closed this issue · 5 comments
@huanghoujing hi, I run the train.py but in the 19 lines :“from aligned_reid.dataset import create_dataset”，but i can't find this file in "dataset" .
Thank you for your attention! This create_dataset is defined in file aligned_reid/dataset/__init__.py.
@huanghoujing
Ok ，I have foud it ！
Hi，houjing，I have run it in my project to test it Whether it in a real time ，but I test it and print time ，it use more time in creating the feate, use time 804ms.
my system :1050ti,7700HK,win10 ,cuda9.0 ，cudnn7.1，python3.6, torch0.4
is that my computer too low？
this is the creat feat code:
ims_var = Variable(transer_var_tensor(torch.from_numpy(images), self.device_id).float(), volatile=True)
feats,_ = self.model(ims_var)[:2]
feats = feats.data.cpu().numpy()
could you help me to improve it？
Your code looks alright. Maybe the speed is determined by the machine.
Ok Thks!