MSFFNet

Semantic Segmentation of Agricultural Crop Multispectral Image Using Feature Fusion

This repository contains the code for my paper: Semantic Segmentation of Agricultural Crop Multispectral Image Using Feature Fusion .

Custom dataset

We do not provide custom datasets. sorry.

DATA-DIR

<dataset>
./rgb                 //rgb images
./etc multispectral   //another multispectral images dic ex) blue, green, nir, re ---
./labels              //.
./train.txt
./test.txt            //Train-test split.

Getting started

This code was tested on linux and requires:

• Python 3.8
• conda3 or miniconda3
• CUDA capable GPU (one is enough)

1. Setup environment

conda create --name <your env name> --file environment.yml

2. Run

Train

python train.py

The results will be rendered and put in ./runs/

Inference

python inference.py

The results will be rendered and put in ./result/

Inference example