segmentation_models VS detectron2

Also, I’d suggest considering switching to the segmentation-models library - it provides U-Net models with a variety of pretrained backbones of as encoders. The author also put out a PyTorch version. https://github.com/qubvel/segmentation_models.pytorch https://github.com/qubvel/segmentation_models

I used segmentation-models (https://github.com/qubvel/segmentation_models) to create a deeplabv3+ model. I havent used it in the last 2 months and now i comeback to the same code and cant use it. Getting ModuleNotFoundError: No module named 'segmentation_models_pytorch.deeplabv3'

I'm building some semantic segmentation models off of low-moderate volumes of biomedical images (~500 - 1k images). So far I've done some hyperparameter sweeping (learning rate, transfer learning, architectures, dropout layers) using the Segmentation Models package from qubvel https://github.com/qubvel/segmentation_models but I'm only seeing moderate performance and minimal differences between tested parameters.

So, image segmentation (especially for satellite images) is a known problem. Search for semantic segmentation and unet (a model used for semantic segmentation). Also, if you use tensorflow there is this library which I found useful segmentation models.

I am doing an image segmentation project using https://github.com/qubvel/segmentation_models as the baseline. I was wondering if any of you have tried the latest segmentation models from cvpr papers. If yes, which ones you found to be interesting or actually improve miou. And how difficult/easy it is to implement those?

That is just U-net from this repo, anything aside is slicing images to fit into window and predict call. I measure time of predict() and it is the same as profiler numbers, so definitely my other operations are beyond profiler. C API code is just creating tensors and calling TF_SessionRun plus slice operations with opencv. Can't post code, sorry.

I'm trying to train image segmentation model with transfer learning using https://github.com/qubvel/segmentation_models/.

I don’t do AI professionally but as a hobby, so this may not be the best way. But the way you described, it seems the user maybe taking the picture a bit further away and there may be other objects in the frame. So you may want to look into some sort of segmentation or have bounding box. This could help the user make sure they are looking at documents for the correct machine.

I think something like detectron2 [1] could help. It is Apache2 license, so commercial friendly. That said the pre-trained weights may not be used for commercial purposes, so you’ll want to check on that.

[1] https://github.com/facebookresearch/detectron2

And not enough true positives either. Add more augmentations in the config. Also make sure the config is set correctly, so that Detectron2 isn't skipping background images: https://github.com/facebookresearch/detectron2/issues/80

I followed the setup of https://github.com/facebookresearch/detectron2/tree/main/projects/TensorMask. But still I can not import it. As I can with from detectron2.projects import point_rend easily from PointRend projects

I have to use this config file with the dataloader which is in https://github.com/facebookresearch/detectron2/blob/main/projects/MViTv2/configs/common/coco_loader.py. I figured that i can use cfg.dataloader.train.dataset.names = "my_dataset_train" for this.

I want to use this config file https://github.com/facebookresearch/detectron2/blob/main/projects/MViTv2/configs/mask_rcnn_mvitv2_t_3x.py like the beneath typical way I use a yaml config file. But giving so many errors one after another that, I even failed to count at this point.

Which is built on https://github.com/facebookresearch/detectron2

Click to Learn more...

Detectron2 (GitHub: https://github.com/facebookresearch/detectron2) is a Facebook AI Research library with state-of-the-art object detection and segmentation algorithms in PyTorch.