https://github.com/arbit3rr/unet-aerialsegmentation

A PyTorch implementation of U-Net for aerial imagery semantic segmentation.

https://github.com/arbit3rr/unet-aerialsegmentation

Science Score: 13.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • CITATION.cff file
  • codemeta.json file
    Found codemeta.json file
  • .zenodo.json file
  • DOI references
  • Academic publication links
  • Academic email domains
  • Institutional organization owner
  • JOSS paper metadata
  • Scientific vocabulary similarity
    Low similarity (2.7%) to scientific vocabulary

Keywords

aerial-imagery cross-entropy-loss focal-loss iou-loss pytorch segmentation semantic-segmentation u-net unet
Last synced: 6 months ago · JSON representation

Repository

A PyTorch implementation of U-Net for aerial imagery semantic segmentation.

Basic Info
  • Host: GitHub
  • Owner: arbit3rr
  • License: gpl-3.0
  • Language: Python
  • Default Branch: main
  • Homepage:
  • Size: 42 KB
Statistics
  • Stars: 81
  • Watchers: 3
  • Forks: 24
  • Open Issues: 0
  • Releases: 0
Topics
aerial-imagery cross-entropy-loss focal-loss iou-loss pytorch segmentation semantic-segmentation u-net unet
Created over 4 years ago · Last pushed over 4 years ago
Metadata Files
Readme License

README.md

U-Net for Semantic Segmentation on Unbalanced Aerial Imagery

Note: this repository is still developing but the colab notebook is complete for training and evaluations.

Read the article at Towards data science

Open In Colab

Kaggle Dataset

``` UNet-AerialSegmentation ├── dataloader.py ├── losses.py ├── model.py ├── train.py └── inference.py

```

dataset_sample1

Training

!python train.py --num_epochs 2 --batch 2 --loss focalloss

Owner

  • Name: Amirhossein Heydarian Ardakani
  • Login: arbit3rr
  • Kind: user

GitHub Events

Total
  • Watch event: 4
  • Fork event: 2
Last Year
  • Watch event: 4
  • Fork event: 2