xdbai-source / Rice-Plant-Counting

Source Codes of Rice Plant Counting
22 stars 1 forks source link

detect.py 存在问题 #17

Open fan2077 opened 2 months ago

fan2077 commented 2 months ago

在运行detect.py时候这块 pre_dis = model_size(images) label = h5py.File(img_path[0].replace('jpg','h5').replace('imgs_4','tep_density'), 'r') predict = torch.from_numpy(np.array(label['density'], dtype=np.float32)) count, kpoint = Center_counting(images, predict, dis, img_path) 代码存在问题,下载的数据集中并没有tep_density,我将它修改为new_data_4,但是代码运行上还是有很多问题

linghanwu-code commented 1 month ago

在运行detect.py时候这块 pre_dis = model_size(images) label = h5py.File(img_path[0].replace('jpg','h5').replace('imgs_4','tep_density'), 'r') predict = torch.from_numpy(np.array(label['density'], dtype=np.float32)) count, kpoint = Center_counting(images, predict, dis, img_path) 代码存在问题,下载的数据集中并没有tep_density,我将它修改为new_data_4,但是代码运行上还是有很多问题

解决了吗?