Skip to content

Commit beaf87c

Browse files
yixu.cuiyixu.cui
authored andcommitted
Make two small changes
1 parent 580d2ff commit beaf87c

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

models/yolo.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -344,6 +344,7 @@ def forward_once(self, x, profile=False):
344344
# [cui] Ref1.0: https://github.com/WongKinYiu/yolov7/issues/647#issuecomment-1238006745
345345
# [cui] Ref1.1: https://github.com/WongKinYiu/yolov7/issues/520#issuecomment-1223438537
346346
# [cui] Ref2: https://github.com/WongKinYiu/yolov7/issues/647#issuecomment-1423777996
347+
# [cui] Ref3: https://github.com/ultralytics/yolov5/issues/6948#issuecomment-1299936326
347348
if isinstance(m, nn.Upsample):
348349
# if check_version(torch.__version__, '1.10.0'):
349350
m.recompute_scale_factor = False

tracker/track_demo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -280,7 +280,7 @@ def save_videos(obj_name):
280280
frame = cv2.imread(os.path.join(images_path, img))
281281
vw.write(frame)
282282

283-
print('Save videos Done!!')
283+
print(f"Save videos to {to_video_path} !!!")
284284

285285

286286
def get_color(idx):

0 commit comments

Comments
 (0)