未验证 提交 d1c6c1e1 编写于 作者: G George Ni 提交者: GitHub

fix mot typo (#3953)

上级 32b1929c
...@@ -251,8 +251,7 @@ def predict_video(detector, camera_id): ...@@ -251,8 +251,7 @@ def predict_video(detector, camera_id):
if not os.path.exists(save_dir): if not os.path.exists(save_dir):
os.makedirs(save_dir) os.makedirs(save_dir)
cv2.imwrite( cv2.imwrite(
os.path.join(save_dir, '{:05d}.jpg'.format(frame_id)), os.path.join(save_dir, '{:05d}.jpg'.format(frame_id)), im)
im)
else: else:
writer.write(im) writer.write(im)
frame_id += 1 frame_id += 1
......
...@@ -392,8 +392,7 @@ def predict_video(detector, reid_model, camera_id): ...@@ -392,8 +392,7 @@ def predict_video(detector, reid_model, camera_id):
if not os.path.exists(save_dir): if not os.path.exists(save_dir):
os.makedirs(save_dir) os.makedirs(save_dir)
cv2.imwrite( cv2.imwrite(
os.path.join(save_dir, '{:05d}.jpg'.format(frame_id)), os.path.join(save_dir, '{:05d}.jpg'.format(frame_id)), im)
im)
else: else:
writer.write(im) writer.write(im)
frame_id += 1 frame_id += 1
......
...@@ -229,4 +229,3 @@ def get_current_memory_mb(): ...@@ -229,4 +229,3 @@ def get_current_memory_mb():
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle) meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
gpu_mem = meminfo.used / 1024. / 1024. gpu_mem = meminfo.used / 1024. / 1024.
return round(cpu_mem, 4), round(gpu_mem, 4), round(gpu_percent, 4) return round(cpu_mem, 4), round(gpu_mem, 4), round(gpu_percent, 4)
...@@ -41,4 +41,3 @@ fetch_map = client.predict( ...@@ -41,4 +41,3 @@ fetch_map = client.predict(
print(fetch_map) print(fetch_map)
fetch_map["image"] = sys.argv[1] fetch_map["image"] = sys.argv[1]
postprocess(fetch_map) postprocess(fetch_map)
...@@ -4,4 +4,3 @@ ...@@ -4,4 +4,3 @@
- [FAQ:第零期](./FAQ第零期.md) - [FAQ:第零期](./FAQ第零期.md)
- [FAQ:第一期](./FAQ第一期.md) - [FAQ:第一期](./FAQ第一期.md)
...@@ -108,7 +108,9 @@ class LetterBoxResize(BaseOperator): ...@@ -108,7 +108,9 @@ class LetterBoxResize(BaseOperator):
if not isinstance(im, np.ndarray): if not isinstance(im, np.ndarray):
raise TypeError("{}: image type is not numpy.".format(self)) raise TypeError("{}: image type is not numpy.".format(self))
if len(im.shape) != 3: if len(im.shape) != 3:
raise ImageError('{}: image is not 3-dimensional.'.format(self)) from PIL import UnidentifiedImageError
raise UnidentifiedImageError(
'{}: image is not 3-dimensional.'.format(self))
# apply image # apply image
height, width = self.target_size height, width = self.target_size
...@@ -135,7 +137,7 @@ class MOTRandomAffine(BaseOperator): ...@@ -135,7 +137,7 @@ class MOTRandomAffine(BaseOperator):
Args: Args:
degrees (list[2]): the rotate range to apply, transform range is [min, max] degrees (list[2]): the rotate range to apply, transform range is [min, max]
translate (list[2]): the translate range to apply, ransform range is [min, max] translate (list[2]): the translate range to apply, transform range is [min, max]
scale (list[2]): the scale range to apply, transform range is [min, max] scale (list[2]): the scale range to apply, transform range is [min, max]
shear (list[2]): the shear range to apply, transform range is [min, max] shear (list[2]): the shear range to apply, transform range is [min, max]
borderValue (list[3]): value used in case of a constant border when appling borderValue (list[3]): value used in case of a constant border when appling
......
...@@ -17,6 +17,7 @@ from __future__ import division ...@@ -17,6 +17,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import os import os
import sys
import copy import copy
import time import time
......
...@@ -357,7 +357,7 @@ def boxes_iou(boxes1, boxes2): ...@@ -357,7 +357,7 @@ def boxes_iou(boxes1, boxes2):
''' '''
area1 = box_area(boxes1) area1 = box_area(boxes1)
area2 = box_area(boxes2) area2 = box_area(boxes2)
lt = paddle.maximum(boxes1.unsqueeze(-2)[:, :, :2], boxes2[:, :2]) lt = paddle.maximum(boxes1.unsqueeze(-2)[:, :, :2], boxes2[:, :2])
rb = paddle.minimum(boxes1.unsqueeze(-2)[:, :, 2:], boxes2[:, 2:]) rb = paddle.minimum(boxes1.unsqueeze(-2)[:, :, 2:], boxes2[:, 2:])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册