|
|
|
@ -1106,7 +1106,8 @@ class Exporter: |
|
|
|
|
raise ValueError("MCT export is not supported for end2end models.") |
|
|
|
|
if "C2f" not in self.model.__str__(): |
|
|
|
|
raise ValueError("MCT export is only supported for YOLOv8 detection models") |
|
|
|
|
check_requirements("model-compression-toolkit==2.1.1") |
|
|
|
|
check_requirements(("model-compression-toolkit==2.1.1", "sony-custom-layers[torch]")) |
|
|
|
|
from sony_custom_layers.pytorch.object_detection.nms import multiclass_nms |
|
|
|
|
import model_compression_toolkit as mct |
|
|
|
|
|
|
|
|
|
def representative_dataset_gen(dataloader=self.get_int8_calibration_dataloader(prefix)): |
|
|
|
@ -1145,10 +1146,6 @@ class Exporter: |
|
|
|
|
)[0] |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
if self.args.nms: |
|
|
|
|
check_requirements("sony-custom-layers[torch]") |
|
|
|
|
from sony_custom_layers.pytorch.object_detection.nms import multiclass_nms |
|
|
|
|
|
|
|
|
|
class NMSWrapper(torch.nn.Module): |
|
|
|
|
def __init__( |
|
|
|
|
self, |
|
|
|
|