| 143 | print('%d images loaded' % self.total) |
| 144 | |
| 145 | def loadImage(self): |
| 146 | |
| 147 | query = self.queries[self.cur - 1] |
| 148 | |
| 149 | rank_list = [] |
| 150 | |
| 151 | rank_list.append(os.path.join(DATA_PATH, 'image_query', query)) |
| 152 | |
| 153 | for gallery in self.rank_pkl[query]: |
| 154 | rank_list.append(os.path.join(DATA_PATH, 'image_test', gallery)) |
| 155 | |
| 156 | self.tmp = [] |
| 157 | self.prbList = [] |
| 158 | |
| 159 | # load query image |
| 160 | f = rank_list[0] + ".jpg" |
| 161 | # print("query_name:", f) |
| 162 | gn = gallery_num[query[0:4]] |
| 163 | self.info.config(text="gallery size: %d"%(gn)) |
| 164 | im = cv2.imread(f) |
| 165 | #cv2.imwrite('test.jpg', im) |
| 166 | query_id = f[-24:-20] |
| 167 | print(np.asarray(im)) |
| 168 | im = cv2.cvtColor(np.asarray(im), cv2.COLOR_BGR2RGB) |
| 169 | im = Image.fromarray(im).resize((200, 200), Image.BILINEAR) |
| 170 | #im = Image.fromarray(im).resize((200, 200), Image.ANTIALIAS) |
| 171 | |
| 172 | im = np.asarray(im) #cv2.cvtColor(np.asarray(im), cv2.COLOR_BGR2RGB) |
| 173 | cv2.putText(im, f[-24:-4], (20,40), cv2.FONT_HERSHEY_COMPLEX,1,(255,0,0),1) |
| 174 | #im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB) |
| 175 | im_h, im_w = im.shape[:2] |
| 176 | if (im_w + im_h) < 100: |
| 177 | thk = 2 |
| 178 | else: |
| 179 | thk = int((im_w + im_h) / 50) |
| 180 | # cv2.rectangle(im, (0, 0), ((im_w-1), (im_h-1)), (0,0,255), thk) |
| 181 | im = Image.fromarray(im) |
| 182 | #im = im.convert('RBG') |
| 183 | r = min(SIZE[0] / im.size[0], SIZE[1] / im.size[1]) |
| 184 | new_size = int(r * im.size[0]), int(r * im.size[1]) |
| 185 | self.tmp.append(im.resize(new_size, Image.ANTIALIAS)) |
| 186 | self.prbList.append(ImageTk.PhotoImage(self.tmp[-1])) |
| 187 | self.prbLabels[0].config(image=self.prbList[-1], width=SIZE[0], height=SIZE[1]) |
| 188 | |
| 189 | # load gallery images |
| 190 | rank_list = rank_list[1:] |
| 191 | siz = len(rank_list) |
| 192 | siz = min(siz, 35) |
| 193 | myrange = np.array(range(siz)) |
| 194 | myrange = myrange + 35 * 0 |
| 195 | myimage = [] |
| 196 | for i in myrange: |
| 197 | print(rank_list[i]) |
| 198 | im = cv2.imread(rank_list[i]+".jpg") |
| 199 | im = cv2.cvtColor(np.asarray(im), cv2.COLOR_BGR2RGB) |
| 200 | im = Image.fromarray(im).resize((200, 200), Image.BILINEAR) |
| 201 | im = np.asarray(im) |
| 202 | gallery_id = rank_list[i][-20:-16] |