JeVoisBase  1.23
JeVois Smart Embedded Machine Vision Toolkit Base Modules
Share this page:
Loading...
Searching...
No Matches
yolo-jevois-export.ModelExporter Class Reference
Inheritance diagram for yolo-jevois-export.ModelExporter:
Collaboration diagram for yolo-jevois-export.ModelExporter:

Public Member Functions

 __init__ (self, yoloModel, device='cpu')
 
 forward (self, x, txt_feats)
 
 export (self, output_dir, model_name, img_width, img_height, num_classes)
 

Public Attributes

 model
 
 device
 

Detailed Description

Definition at line 15 of file yolo-jevois-export.py.

Constructor & Destructor Documentation

◆ __init__()

yolo-jevois-export.ModelExporter.__init__ (   self,
  yoloModel,
  device = 'cpu' 
)

Member Function Documentation

◆ export()

yolo-jevois-export.ModelExporter.export (   self,
  output_dir,
  model_name,
  img_width,
  img_height,
  num_classes 
)

Definition at line 31 of file yolo-jevois-export.py.

References yolo-jevois-export.ModelExporter.device.

◆ forward()

Member Data Documentation

◆ device

yolo-jevois-export.ModelExporter.device

Definition at line 26 of file yolo-jevois-export.py.

Referenced by yolo-jevois-export.ModelExporter.export().

◆ model


The documentation for this class was generated from the following file: