Dickson Neoh

Update black version in setup.cfg and pyproject.toml

1

Updating black version throughout the repo. `black==20.8b1` throws an error like the following: ``` Traceback (most recent call last): File "/opt/hostedtoolcache/Python/3.7.13/x64/bin/black", line 8, in sys.exit(patched_main()) File "/opt/hostedtoolcache/Python/3.7.13/x64/lib/python3.7/site-packages/black/__init__.py", line 6606, in...

We are targeting to integrate FCOS from `torchvision==0.12.0` which depends on `torch==1.11.0`. But at the same time we need `torch==1.10.0` because `mmdet==2.17.0` depends on it. We could not update `mmdet`...

Update to cuda 11.3

1

Update cuda 11.1 to cuda 11.3 which is installed by default on Colab.

Add PVTv2 backbone to RetinaNet

5

This PR adds [Pyramid Vision Transformer v2](https://arxiv.org/abs/2106.13797) backbone to the RetinaNet model. Configs and weights from mmdet repo https://github.com/open-mmlab/mmdetection/tree/master/configs/pvt

Multiple image extension for YOLOv5 dataset

8

My Yolov5 dataset has got multiple image extensions files such as jpg, jpeg, png, etc. What is the best way to handle these cases? ``` dataset = importer.ImportYoloV5(path=path_to_annotations, path_to_images=path_to_images, cat_names=yoloclasses,...

### Contact Details [Optional] _No response_ ### System Information ``` ZenML version: 0.11.0 Install path: /home/dnth/anaconda3/envs/zenfiles/lib/python3.8/site-packages/zenml Python version: 3.8.13 Platform information: {'os': 'linux', 'linux_distro': 'ubuntu', 'linux_distro_like': 'debian', 'linux_distro_version': '20.04'} Environment:...

### Contact Details [Optional] _No response_ ### System Information ``` ZenML version: 0.11.0 Install path: /home/dnth/anaconda3/envs/zenfiles/lib/python3.8/site-packages/zenml Python version: 3.8.13 Platform information: {'os': 'linux', 'linux_distro': 'ubuntu', 'linux_distro_like': 'debian', 'linux_distro_version': '20.04'} Environment:...

[DOCS] Add ZenML into integration table in Readme

2

Changes proposed in this pull request: - Adds ZenML as an open-source ML pipeline tool compatible with Great Expectations.

Export to ONNX supported?

13

I'm trying to export the model into `.onnx`. Here's my code: ```python import torch model = torch.hub.load('facebookresearch/dinov2', 'dinov2_vits14').to('cuda:0') model.eval() # Generate some input data input_data = torch.randn(1, 3, 224, 224).to('cuda:0')...

I'd like to remove only the non-text background. Can this be done? For example, I'd like to remove the background of this image and keep the lyrics text