Zhiwei Chen

Results 2 issues of Zhiwei Chen

RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 'other' added:prediction = prediction.cuda()

hello, I'm afraid that line 1035(https://github.com/lzx1413/labelImgPlus/blob/master/labelImg.py#L1035) and line 1047(https://github.com/lzx1413/labelImgPlus/blob/master/labelImg.py#L1047) cause the issue of IOError mentioned on the title. To solve them, line 1035 should be replaced by `result_path = os.path.join(self.defaultSaveDir,os.path.splitext(imgFileName)[0])...