wkcn / MobulaOP

A Simple & Flexible Cross Framework Operators Toolkit
MIT License
164 stars 21 forks source link

Custom Operators Zoo #87

Open wkcn opened 4 years ago

wkcn commented 4 years ago

Hi there, this issue is to summarize some custom operators to be supported. Please feel free to add it if you want any operator : )

kohillyang commented 4 years ago

I suggest several operators for reference:

  1. CTC Loss, this OP is supported by mxnet, but before PyTorch 1.0, users must compile by themselves.

  2. Deformable Convolution (and also DCNV2), Same as 1, users need compiling, and mxnet does not provide CPU implementation version.

  3. Image wrapAffine based on Homography matrix. If this one is implemented, users can implement their image augmentations like image crop, translation, rotating and even scaling based on GPU.