Deci-AI
/
super-gradients
connected to https://github.com/Deci-AI/super-gradients.git


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
            from super_gradients.training.datasets import CoCoDetectionDatasetInterface
from super_gradients.training.utils.detection_utils import AnchorGenerator

dataset_params = {"batch_size": 1,
                  "test_batch_size": 1,
                  "dataset_dir": "/data/coco/",
                  "image_size": 640,
                  "train_sample_loading_method": 'rectangular'
                  }

coco_dataset_interface = CoCoDetectionDatasetInterface(dataset_params=dataset_params, cache_labels=True)
anchors = AnchorGenerator(coco_dataset_interface.trainset, 9)