xview3-kohei-solution

Usage

This repository assumes that the given data set is stored in the following locations:

$ ls data/input/xview3/*.csv
data/input/xview3/train.csv  data/input/xview3/validation.csv

$ ls data/input/xview3/downloaded
00a035722196ee86t.tar.gz  4455faa0cb4824f4t.tar.gz  85fe34d1aee53a7ft.tar.gz  c07f6ec980c2c149t.tar.gz
014261f774287442t.tar.gz  4518c556b38a5fa4t.tar.gz  864390795b0439b1t.tar.gz  c0831dbd6d7f3c56t.tar.gz
(snip)

In the case of training, create a virtual env as follows:

# Setup: Create the virtual env
$ poetry config virtualenvs.in-project true
$ poetry install
$ poetry run pip install albumentations timm segmentation_models_pytorch torch==1.10.0+cu113 torchvision==0.11.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

Training

To save storage space, my preprocessing code assumes that the input image file is the original file in .tar.gz format. It does NOT assume pre-extracted files. The batch size and other settings are based on the assumption that two RTX3080 GPU cards are used.

$ bash train.sh

Inference

My containerized inference code follows the xView3 evaluation protocol. The detailed usage of the xview3 evaluation protocol is described in https://iuu.xview.us/verify.

To build the Docker Image, you need the model weights generated by train.sh. The pre-computed files can be downloaded from my Google Drive.

You will also need to download and extract gshhg-shp-2.3.7.zip as external data. GSHHG data can be found at https://www.ngdc.noaa.gov/mgg/shorelines/data/gshhg/latest/

The Docker image containing these model weights and external data has already been uploaded to Docker Hub (smly/kohei-xview3:latest).

## Setup: Build the docker image -----
$ docker build --no-cache -t kohei-xview3 .

## Inference -----
$ docker run \
    --shm-size 16G \
    --gpus=1 \
    --mount type=bind,source=/home/xv3data,target=/on-docker/xv3data \
    kohei-xview3 \
    /on-docker/xv3data/ \
    0157baf3866b2cf9v \
    /on-docker/xv3data/prediction/prediction.csv

Kohei's 5th place solution for xview3 challenge

Related tags

Overview

xview3-kohei-solution

Usage

Training

Inference

Owner

Kohei Ozaki

A clean and scalable template to kickstart your deep learning project 🚀 ⚡ 🔥

Nsdf: A mesh SDF with just some code we can directly paste into our raymarcher

PyTorch implementation of the implicit Q-learning algorithm (IQL)

A framework for Quantification written in Python

An educational tool to introduce AI planning concepts using mobile manipulator robots.

Human Activity Recognition example using TensorFlow on smartphone sensors dataset and an LSTM RNN. Classifying the type of movement amongst six activity categories - Guillaume Chevalier

This package proposes simplified exporting pytorch models to ONNX and TensorRT, and also gives some base interface for model inference.

[CVPRW 2021] Code for Region-Adaptive Deformable Network for Image Quality Assessment

FluxTraining.jl gives you an endlessly extensible training loop for deep learning

用opencv的dnn模块做yolov5目标检测，包含C++和Python两个版本的程序

🔥 TensorFlow Code for technical report: "YOLOv3: An Incremental Improvement"

Super-BPD: Super Boundary-to-Pixel Direction for Fast Image Segmentation (CVPR 2020)

Anti-UAV base on PaddleDetection

Code base for the paper "Scalable One-Pass Optimisation of High-Dimensional Weight-Update Hyperparameters by Implicit Differentiation"

Build Graph Nets in Tensorflow

Depression Asisstant GDSC Challenge Solution

Semi-supervised Implicit Scene Completion from Sparse LiDAR

Learning with Noisy Labels via Sparse Regularization, ICCV2021

Implementation of "Bidirectional Projection Network for Cross Dimension Scene Understanding" CVPR 2021 (Oral)

Equivariant GNN for the prediction of atomic multipoles up to quadrupoles.