Skip to content
This repository has been archived by the owner on Jul 15, 2023. It is now read-only.

Team-BoonMoSa/PANPP

Repository files navigation

Model Structure

Model

Directory Structure
├── cfg.py
├── checkpoints
│   └── cfg
│       └── checkpoint.pth.tar
├── data
│   ├── TestData
│   │   ├── image
│   │   └── txt
│   └── TrainingData
│       ├── image
│       └── txt
├── dataset
│   ├── builder.py
│   ├── __init__.py
│   ├── pan_pp_test.py
│   └── pan_pp_train.py
├── models
│   ├── backbone
│   │   ├── builder.py
│   │   ├── __init__.py
│   │   └── resnet.py
│   ├── builder.py
│   ├── head
│   │   ├── builder.py
│   │   ├── __init__.py
│   │   └── pan_pp_det_head.py
│   ├── __init__.py
│   ├── loss
│   │   ├── acc.py
│   │   ├── builder.py
│   │   ├── dice_loss.py
│   │   ├── emb_loss_v1.py
│   │   ├── emb_loss_v2.py
│   │   ├── __init__.py
│   │   ├── iou.py
│   │   └── ohem.py
│   ├── neck
│   │   ├── builder.py
│   │   ├── fpem_v2.py
│   │   └── __init__.py
│   ├── pan_pp.py
│   └── utils
│       ├── conv_bn_relu.py
│       ├── coordconv.py
│       ├── fuse_conv_bn.py
│       └── __init__.py
├── README.md
├── requirement.txt
├── main.ipynb
├── test.py
├── test.sh
├── train.py
├── train.sh
└── utils
    ├── average_meter.py
    ├── corrector.py
    ├── __init__.py
    ├── logger.py
    ├── result_format.py
    └── visualizer.py

Init

~/Google Drive/내 드라이브$ git clone https://github.com/Team-BoonMoSa/PANPP

~/Google Drive/내 드라이브/PANPP/dataFlickrLogos-v2 업로드


Make Ground Truth

sh makeGT.sh

Train

!sh train.sh

Prepare Training Data

./dataset/pan_pp_train.py

tt_root_dir = './data/TrainingData/'
tt_train_data_dir = tt_root_dir + 'image/'
tt_train_gt_dir = tt_root_dir + 'txt/'

Fine-Tuning

./cfg.py

train_cfg = dict(
    ...
    # pretrain='',
)

Test

!sh test.sh

Prepare Test Data

./cfg.py

data = dict(
    ...
    test=dict(
        ...
        data='data/TestData/image/'
    )
)

Prepare Weights of Pretrained PAN++

./cfg.py

test_cfg = dict(
    ...
    pretrain='./checkpoints/cfg/checkpoint.pth.tar', # tmp
)

Reference

About

No description or website provided.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published