Skip to content

Commit

Permalink
Merge pull request #705 from roboflow/feature/flicker-killer
Browse files Browse the repository at this point in the history
Add StabilizeTrackedDetectionsBlock
  • Loading branch information
PawelPeczek-Roboflow authored Oct 17, 2024
2 parents 90dfb26 + f8585f8 commit 1fa4c96
Show file tree
Hide file tree
Showing 4 changed files with 537 additions and 66 deletions.
136 changes: 70 additions & 66 deletions inference/core/workflows/core_steps/loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -201,6 +201,9 @@
from inference.core.workflows.core_steps.transformations.relative_static_crop.v1 import (
RelativeStaticCropBlockV1,
)
from inference.core.workflows.core_steps.transformations.stabilize_detections.v1 import (
StabilizeTrackedDetectionsBlockV1,
)
from inference.core.workflows.core_steps.transformations.stitch_images.v1 import (
StitchImagesBlockV1,
)
Expand Down Expand Up @@ -308,93 +311,94 @@

def load_blocks() -> List[Type[WorkflowBlock]]:
return [
TimeInZoneBlockV1,
BoundingRectBlockV1,
SegmentAnything2BlockV1,
DetectionsConsensusBlockV1,
ClipComparisonBlockV1,
LMMBlockV1,
LMMForClassificationBlockV1,
OpenAIBlockV1,
CogVLMBlockV1,
OCRModelBlockV1,
YoloWorldModelBlockV1,
RoboflowInstanceSegmentationModelBlockV1,
RoboflowKeypointDetectionModelBlockV1,
RoboflowClassificationModelBlockV1,
RoboflowMultiLabelClassificationModelBlockV1,
RoboflowObjectDetectionModelBlockV1,
BarcodeDetectorBlockV1,
QRCodeDetectorBlockV1,
AbsoluteStaticCropBlockV1,
DynamicCropBlockV1,
DetectionsFilterBlockV1,
DetectionOffsetBlockV1,
ByteTrackerBlockV1,
RelativeStaticCropBlockV1,
DetectionsTransformationBlockV1,
RoboflowDatasetUploadBlockV1,
ContinueIfBlockV1,
PerspectiveCorrectionBlockV1,
DynamicZonesBlockV1,
SizeMeasurementBlockV1,
DetectionsClassesReplacementBlockV1,
ExpressionBlockV1,
PropertyDefinitionBlockV1,
DimensionCollapseBlockV1,
FirstNonEmptyOrDefaultBlockV1,
AntropicClaudeBlockV1,
BackgroundColorVisualizationBlockV1,
BarcodeDetectorBlockV1,
BlurVisualizationBlockV1,
BoundingBoxVisualizationBlockV1,
BoundingRectBlockV1,
ByteTrackerBlockV1,
ByteTrackerBlockV2,
CameraFocusBlockV1,
CircleVisualizationBlockV1,
ClipComparisonBlockV1,
ClipComparisonBlockV2,
CogVLMBlockV1,
ColorVisualizationBlockV1,
ContinueIfBlockV1,
ConvertGrayscaleBlockV1,
CornerVisualizationBlockV1,
CropVisualizationBlockV1,
DetectionOffsetBlockV1,
DetectionsClassesReplacementBlockV1,
DetectionsConsensusBlockV1,
DetectionsFilterBlockV1,
DetectionsStitchBlockV1,
DetectionsTransformationBlockV1,
DimensionCollapseBlockV1,
DistanceMeasurementBlockV1,
DominantColorBlockV1,
DotVisualizationBlockV1,
DynamicCropBlockV1,
DynamicZonesBlockV1,
EllipseVisualizationBlockV1,
ExpressionBlockV1,
FirstNonEmptyOrDefaultBlockV1,
Florence2BlockV1,
GoogleGeminiBlockV1,
GoogleVisionOCRBlockV1,
HaloVisualizationBlockV1,
ImageBlurBlockV1,
ImageContoursDetectionBlockV1,
ImagePreprocessingBlockV1,
ImageSlicerBlockV1,
ImageThresholdBlockV1,
JSONParserBlockV1,
LMMBlockV1,
LMMForClassificationBlockV1,
LabelVisualizationBlockV1,
LineCounterBlockV1,
LineCounterBlockV2,
LineCounterZoneVisualizationBlockV1,
MaskVisualizationBlockV1,
ModelComparisonVisualizationBlockV1,
OCRModelBlockV1,
OpenAIBlockV1,
OpenAIBlockV2,
PathDeviationAnalyticsBlockV1,
PathDeviationAnalyticsBlockV2,
PerspectiveCorrectionBlockV1,
PixelateVisualizationBlockV1,
PixelationCountBlockV1,
PolygonVisualizationBlockV1,
LineCounterZoneVisualizationBlockV1,
ModelComparisonVisualizationBlockV1,
TriangleVisualizationBlockV1,
PolygonZoneVisualizationBlockV1,
PropertyDefinitionBlockV1,
QRCodeDetectorBlockV1,
RelativeStaticCropBlockV1,
RoboflowClassificationModelBlockV1,
RoboflowCustomMetadataBlockV1,
DetectionsStitchBlockV1,
ImageSlicerBlockV1,
DominantColorBlockV1,
PixelationCountBlockV1,
RoboflowDatasetUploadBlockV1,
RoboflowDatasetUploadBlockV2,
RoboflowInstanceSegmentationModelBlockV1,
RoboflowKeypointDetectionModelBlockV1,
RoboflowMultiLabelClassificationModelBlockV1,
RoboflowObjectDetectionModelBlockV1,
SIFTBlockV1,
SIFTComparisonBlockV1,
SIFTComparisonBlockV2,
SIFTBlockV1,
TemplateMatchingBlockV1,
ImageBlurBlockV1,
ConvertGrayscaleBlockV1,
ImageThresholdBlockV1,
ImageContoursDetectionBlockV1,
ClipComparisonBlockV2,
CameraFocusBlockV1,
RoboflowDatasetUploadBlockV2,
SegmentAnything2BlockV1,
SizeMeasurementBlockV1,
StabilityAIInpaintingBlockV1,
StabilizeTrackedDetectionsBlockV1,
StitchImagesBlockV1,
OpenAIBlockV2,
JSONParserBlockV1,
TemplateMatchingBlockV1,
TimeInZoneBlockV1,
TimeInZoneBlockV2,
TriangleVisualizationBlockV1,
VLMAsClassifierBlockV1,
GoogleGeminiBlockV1,
GoogleVisionOCRBlockV1,
VLMAsDetectorBlockV1,
AntropicClaudeBlockV1,
LineCounterBlockV1,
PolygonZoneVisualizationBlockV1,
Florence2BlockV1,
DistanceMeasurementBlockV1,
StabilityAIInpaintingBlockV1,
ImagePreprocessingBlockV1,
PathDeviationAnalyticsBlockV1,
ByteTrackerBlockV2,
PathDeviationAnalyticsBlockV2,
TimeInZoneBlockV2,
LineCounterBlockV2,
YoloWorldModelBlockV1,
]


Expand Down
Empty file.
Loading

0 comments on commit 1fa4c96

Please sign in to comment.