Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

#468 Bug/sg 399 external checkpoints fix

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:bug/SG-399_external_checkpoints_fix
1
2
3
4
5
6
7
8
9
10
  1. # PACKAGE IMPORTS FOR EXTERNAL USAGE
  2. import cv2
  3. from super_gradients.training.transforms.transforms import DetectionMosaic, DetectionRandomAffine, DetectionHSV,\
  4. DetectionPaddedRescale, DetectionTargetsFormatTransform
  5. from super_gradients.training.transforms.all_transforms import TRANSFORMS, Transforms
  6. __all__ = ['TRANSFORMS', 'Transforms', 'DetectionMosaic', 'DetectionRandomAffine', 'DetectionHSV', 'DetectionPaddedRescale',
  7. 'DetectionTargetsFormatTransform']
  8. cv2.setNumThreads(0)
Discard
Tip!

Press p or to see the previous file or, n or to see the next file