Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

优化修正文档检测中的部分语法

Browse Source
This commit is contained in:
liuyebo
2024-08-20 14:27:51 +08:00
parent 299b762cad
commit 896d2aaf9b
3 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
object_detection/core/test/cornernet_saccade.py
View File

@@ -233,7 +233,7 @@ def prepare_images(db, image, locs, flipped=True):
input_size = db.configs["input_size"]
num_patches = locs.shape[0]
images = torch.cuda.FloatTensor(num_patches, 3, input_size[0], input_size[1]).fill_(0)
images = torch.zeros((num_patches, 3, input_size[0], input_size[1]), dtype=torch.float32, device='cuda')
offsets = np.zeros((num_patches, 2), dtype=np.float32)
for ind, (y, x, scale) in enumerate(locs[:, :3]):
crop_height = int(input_size[0] / scale)
@@ -319,10 +319,9 @@ def cornernet_saccade_inference(db, nnet, image, decode_func=batch_decode):
num_iterations = len(att_thresholds)
im_mean = torch.cuda.FloatTensor(db.mean).reshape(1, 3, 1, 1)
im_std = torch.cuda.FloatTensor(db.std).reshape(1, 3, 1, 1)
im_mean = torch.tensor(db.mean, dtype=torch.float32, device='cuda').reshape(1, 3, 1, 1)
im_std = torch.tensor(db.std, dtype=torch.float32, device='cuda').reshape(1, 3, 1, 1)
detections = []
height, width = image.shape[0:2]
image = image / 255.