ternaus / Midv 500 Models
Licence: mit
Model for document segmentation trained on the midv-500-models dataset.
Stars: ✭ 31
Programming Languages
python
139335 projects - #7 most used programming language
Projects that are alternatives of or similar to Midv 500 Models
FCN-Segmentation-TensorFlow
FCN for Semantic Image Segmentation achieving 68.5 mIoU on PASCAL VOC
Stars: ✭ 34 (+9.68%)
Mutual labels: image-segmentation, semantic-segmentation
Segmentation models.pytorch
Segmentation models with pretrained backbones. PyTorch.
Stars: ✭ 4,584 (+14687.1%)
Mutual labels: semantic-segmentation, image-segmentation
DLCV2018SPRING
Deep Learning for Computer Vision (CommE 5052) in NTU
Stars: ✭ 38 (+22.58%)
Mutual labels: image-segmentation, semantic-segmentation
BMW-IntelOpenVINO-Segmentation-Inference-API
This is a repository for a semantic segmentation inference API using the OpenVINO toolkit
Stars: ✭ 31 (+0%)
Mutual labels: image-segmentation, semantic-segmentation
Refinenet
RefineNet: Multi-Path Refinement Networks for High-Resolution Semantic Segmentation
Stars: ✭ 543 (+1651.61%)
Mutual labels: semantic-segmentation, image-segmentation
InstantDL
InstantDL: An easy and convenient deep learning pipeline for image segmentation and classification
Stars: ✭ 33 (+6.45%)
Mutual labels: image-segmentation, semantic-segmentation
Geospatial Machine Learning
A curated list of resources focused on Machine Learning in Geospatial Data Science.
Stars: ✭ 289 (+832.26%)
Mutual labels: semantic-segmentation, image-segmentation
K-Net
[NeurIPS2021] Code Release of K-Net: Towards Unified Image Segmentation
Stars: ✭ 434 (+1300%)
Mutual labels: image-segmentation, semantic-segmentation
Pytorch Unet
Simple PyTorch implementations of U-Net/FullyConvNet (FCN) for image segmentation
Stars: ✭ 470 (+1416.13%)
Mutual labels: semantic-segmentation, image-segmentation
Pixellib
Visit PixelLib's official documentation https://pixellib.readthedocs.io/en/latest/
Stars: ✭ 327 (+954.84%)
Mutual labels: semantic-segmentation, image-segmentation
SUIM
Semantic Segmentation of Underwater Imagery: Dataset and Benchmark. #IROS2020
Stars: ✭ 53 (+70.97%)
Mutual labels: image-segmentation, semantic-segmentation
Efficient Segmentation Networks
Lightweight models for real-time semantic segmentationon PyTorch (include SQNet, LinkNet, SegNet, UNet, ENet, ERFNet, EDANet, ESPNet, ESPNetv2, LEDNet, ESNet, FSSNet, CGNet, DABNet, Fast-SCNN, ContextNet, FPENet, etc.)
Stars: ✭ 579 (+1767.74%)
Mutual labels: semantic-segmentation, image-segmentation
Semantic-Segmentation-BiSeNet
Keras BiseNet architecture implementation
Stars: ✭ 55 (+77.42%)
Mutual labels: image-segmentation, semantic-segmentation
Entity
EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation
Stars: ✭ 313 (+909.68%)
Mutual labels: image-segmentation, semantic-segmentation
RGBD-semantic-segmentation
A paper list of RGBD semantic segmentation (processing)
Stars: ✭ 264 (+751.61%)
Mutual labels: image-segmentation, semantic-segmentation
TensorFlow-Advanced-Segmentation-Models
A Python Library for High-Level Semantic Segmentation Models based on TensorFlow and Keras with pretrained backbones.
Stars: ✭ 64 (+106.45%)
Mutual labels: image-segmentation, semantic-segmentation
Smoothly Blend Image Patches
Using a U-Net for image segmentation, blending predicted patches smoothly is a must to please the human eye.
Stars: ✭ 218 (+603.23%)
Mutual labels: semantic-segmentation, image-segmentation
ResUNetPlusPlus-with-CRF-and-TTA
ResUNet++, CRF, and TTA for segmentation of medical images (IEEE JBIHI)
Stars: ✭ 98 (+216.13%)
Mutual labels: image-segmentation, semantic-segmentation
Kaggle salt bes phalanx
Winning solution for the Kaggle TGS Salt Identification Challenge.
Stars: ✭ 317 (+922.58%)
Mutual labels: semantic-segmentation, image-segmentation
Crfasrnn keras
CRF-RNN Keras/Tensorflow version
Stars: ✭ 576 (+1758.06%)
Mutual labels: semantic-segmentation, image-segmentation
midv-500-models
The repository contains a model for binary semantic segmentation of the documents.
- Left: input.
- Center: prediction.
- Right: overlay of the image and predicted mask.
Installation
pip install -U midv500models
Example inference
Jupyter notebook with an example:
Dataset
Model is trained on MIDV-500: A Dataset for Identity Documents Analysis and Recognition on Mobile Devices in Video Stream.
Preparation
Download the dataset from the ftp server with
wget -r ftp://smartengines.com/midv-500/
Unpack the dataset
cd smartengines.com/midv-500/dataset/
unzip \*.zip
The resulting folder structure will be
smartengines.com
midv-500
dataset
01_alb_id
ground_truth
CA
CA01_01.tif
...
images
CA
CA01_01.json
...
...
...
...
...
To preprocess the data use the script
python midv500models/preprocess_data.py -i <input_folder> \
-o <output_folder>
where input_folder
corresponds to the file with the unpacked dataset and output folder will look as:
images
CA01_01.jpg
...
masks
CA01_01.png
target binary masks will have values [0, 255], where 0 is background and 255 is the document.
Training
python midv500models/train.py -c midv500models/configs/2020-05-19.yaml \
-i <path to train>
Inference
python midv500models/inference.py -c midv500models/configs/2020-05-19.yaml \
-i <path to images> \
-o <path to save preidctions>
-w <path to weights>
Weights
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].