diff --git a/docs/en/reference/nn/modules/block.md b/docs/en/reference/nn/modules/block.md index 8aac2580..f8b1e581 100644 --- a/docs/en/reference/nn/modules/block.md +++ b/docs/en/reference/nn/modules/block.md @@ -78,3 +78,11 @@ keywords: YOLO, Ultralytics, neural network, nn.modules.block, Proto, HGBlock, S ## ::: ultralytics.nn.modules.block.BottleneckCSP

+ +## ::: ultralytics.nn.modules.block.ResNetBlock + +

+ +## ::: ultralytics.nn.modules.block.ResNetLayer + +

diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index ed780b1e..0363665e 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = '8.0.224' +__version__ = '8.0.225' from ultralytics.models import RTDETR, SAM, YOLO from ultralytics.models.fastsam import FastSAM diff --git a/ultralytics/engine/predictor.py b/ultralytics/engine/predictor.py index 83abbd2a..5e04b179 100644 --- a/ultralytics/engine/predictor.py +++ b/ultralytics/engine/predictor.py @@ -345,9 +345,9 @@ class BasePredictor: else: # 'video' or 'stream' frames_path = f'{save_path.split(".", 1)[0]}_frames/' if self.vid_path[idx] != save_path: # new video + self.vid_path[idx] = save_path if self.args.save_frames: Path(frames_path).mkdir(parents=True, exist_ok=True) - self.vid_path[idx] = save_path self.vid_frame[idx] = 0 if isinstance(self.vid_writer[idx], cv2.VideoWriter): self.vid_writer[idx].release() # release previous video writer diff --git a/ultralytics/solutions/heatmap.py b/ultralytics/solutions/heatmap.py index 80e8899f..9ef99ca5 100644 --- a/ultralytics/solutions/heatmap.py +++ b/ultralytics/solutions/heatmap.py @@ -20,16 +20,16 @@ class Heatmap: def __init__(self): """Initializes the heatmap class with default values for Visual, Image, track, count and heatmap parameters.""" - # Visual Information + # Visual information self.annotator = None self.view_img = False - # Image Information + # Image information self.imw = None self.imh = None self.im0 = None - # Heatmap Colormap and heatmap np array + # Heatmap colormap and heatmap np array self.colormap = None self.heatmap = None self.heatmap_alpha = 0.5 @@ -40,7 +40,7 @@ class Heatmap: self.clss = None self.track_history = None - # Counting Info + # Counting info self.count_reg_pts = None self.count_region = None self.in_counts = 0 @@ -160,7 +160,8 @@ class Heatmap: return im0_with_heatmap - def display_frames(self, im0_with_heatmap): + @staticmethod + def display_frames(im0_with_heatmap): """ Display heatmap.