Spaces:

koajoel
/

PolyFormer

Running

App Files Files Community

jiang commited on May 16, 2023

Commit

3c6babc

•

1 Parent(s): 650c5f6

update

Browse files

Files changed (1) hide show

README.md +13 -185

README.md CHANGED Viewed

@@ -1,185 +1,13 @@
-# PolyFormer: Referring Image Segmentation as Sequential Polygon Generation (CVPR 2023)
-[![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/polyformer-referring-image-segmentation-as/referring-expression-segmentation-on-refcocog)](https://paperswithcode.com/sota/referring-expression-segmentation-on-refcocog?p=polyformer-referring-image-segmentation-as)
-[![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/polyformer-referring-image-segmentation-as/referring-expression-segmentation-on-refcoco)](https://paperswithcode.com/sota/referring-expression-segmentation-on-refcoco?p=polyformer-referring-image-segmentation-as)
-[![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/polyformer-referring-image-segmentation-as/referring-expression-segmentation-on-refcoco-1)](https://paperswithcode.com/sota/referring-expression-segmentation-on-refcoco-1?p=polyformer-referring-image-segmentation-as)
-[![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/polyformer-referring-image-segmentation-as/referring-expression-comprehension-on-refcoco)](https://paperswithcode.com/sota/referring-expression-comprehension-on-refcoco?p=polyformer-referring-image-segmentation-as)
-[![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/polyformer-referring-image-segmentation-as/referring-expression-comprehension-on-refcoco-1)](https://paperswithcode.com/sota/referring-expression-comprehension-on-refcoco-1?p=polyformer-referring-image-segmentation-as)
-\[[Project Page](https://polyformer.github.io/)\]   \[[Paper](https://arxiv.org/abs/2302.07387)\]
-by [Jiang Liu*](https://joellliu.github.io/), [Hui Ding*](http://www.huiding.org/), [Zhaowei Cai](https://zhaoweicai.github.io/),  [Yuting Zhang](https://scholar.google.com/citations?user=9UfZJskAAAAJ&hl=en), [Ravi Kumar Satzoda](https://scholar.google.com.sg/citations?user=4ngycwIAAAAJ&hl=en), [Vijay Mahadevan](https://scholar.google.com/citations?user=n9fRgvkAAAAJ&hl=en), [R. Manmatha](https://ciir.cs.umass.edu/~manmatha/).
-## :notes: Introduction
-![github_figure](pipeline.gif)
-PolyFormer is a unified model for referring image segmentation (polygon vertex sequence) and referring expression comprehension (bounding box corner points). The polygons are converted to segmentation masks in the end.
-**Contributions:**
-* State-of-the-art results on referring image segmentation and referring expression comprehension on 6 datasets;
-* A unified framework for referring image segmentation (RIS) and referring expression comprehension (REC) by formulating them as a sequence-to-sequence (seq2seq) prediction problem;
-* A regression-based decoder for accurate coordinate prediction, which outputs continuous 2D coordinates directly without quantization error..
-## Getting Started
-### Installation
-```bash
-conda create -n polyformer python=3.7.4
-conda activate polyformer
-python -m pip install -r requirements.txt
-```
-Note: if you are getting import errors from `fairseq`, try the following:
-```bash
-python -m pip install pip==21.2.4
-pip uninstall fairseq
-pip install -r requirements.txt
-```
-## Datasets
-### Prepare Pretraining Data
-1. Create the dataset folders
-```bash
-mkdir datasets
-mkdir datasets/images
-mkdir datasets/annotations
-```
-2. Download the *2014 Train images [83K/13GB]* from [COCO](https://cocodataset.org/#download),
-original [Flickr30K images](http://shannon.cs.illinois.edu/DenotationGraph/),
-[ReferItGame images](https://drive.google.com/file/d/1R6Tm7tQTHCil6A_eOhjudK3rgaBxkD2t/view?usp=sharing),
-and [Visual Genome images](http://visualgenome.org/api/v0/api_home.html), and extract them to `datasets/images`.
-3. Download the annotation file for pretraining datasets [instances.json](https://drive.google.com/drive/folders/1O4hzL8_s3aUsnj_JZnM3CwANd7TejcJO)
-provided by [SeqTR](https://github.com/sean-zhuh/SeqTR) and store it in `datasets/annotations`.
-The workspace directory should be organized like this:
-```
-PolyFormer/
-├── datasets/
-│   ├── images
-│   │   ├── flickr30k/*.jpg
-│   │   ├── mscoco/
-│   │   │   └── train2014/*.jpg
-│   │   ├── saiaprtc12/*.jpg
-│   │   └── visual-genome/*.jpg
-│   └── annotations
-│       └── instances.json
-└── ...
-```
-4. Generate the tsv files for pretraining
-```bash
-python data/create_pretraining_data.py
-```
-### Prepare Finetuning Data
-1. Follow the instructions in the `./refer` directory to set up subdirectories
-and download annotations.
-This directory is based on the [refer](https://github.com/lichengunc/refer) API.
-2. Generate the tsv files for finetuning
-```bash
-python data/create_finetuning_data.py
-```
-## Pretraining
-1. Create the checkpoints folder
-```bash
-mkdir weights
-```
-2. Download pretrain weights of [Swin-base](https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_base_patch4_window12_384_22k.pth),
-[Swin-large](https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_large_patch4_window12_384_22k.pth),
-[BERT-base](https://cdn.huggingface.co/bert-base-uncased-pytorch_model.bin)
-and put the weight files in `./pretrained_weights`.
-These weights are needed for training to initialize the model.
-3. Run the pretraining scripts for model pretraining on the referring expression comprehension task:
-```bash
-cd run_scripts/pretrain
-bash pretrain_polyformer_b.sh  # for pretraining PolyFormer-B model
-bash pretrain_polyformer_l.sh  # for pretraining PolyFormer-L model
-```
-## Finetuning
-Run the finetuning scripts for model pretraining on the referring image segmentation and referring expression comprehension tasks:
-```bash
-cd run_scripts/finetune
-bash train_polyformer_b.sh  # for finetuning PolyFormer-B model
-bash train_polyformer_l.sh  # for finetuning PolyFormer-L model
-```
-Please make sure to link the pretrain weight paths (Line 20) in the finetuning scripts to the best pretraining checkpoints.
-## Evaluation
-Run the evaluation scripts for evaluating on the referring image segmentation and referring expression comprehension tasks:
-```bash
-cd run_scripts/evaluation
-# for evaluating PolyFormer-B model
-bash evaluate_polyformer_b_refcoco.sh
-bash evaluate_polyformer_b_refcoco+.sh
-bash evaluate_polyformer_b_refcocog.sh
-# for evaluating PolyFormer-L model
-bash evaluate_polyformer_l_refcoco.sh
-bash evaluate_polyformer_l_refcoco+.sh
-bash evaluate_polyformer_l_refcocog.sh
-```
-## Model Zoo
-Download the model weights to `./weights` if you want to use our trained models for finetuning and evaluation.
-|                                                                                                       | Refcoco val|           | | Refcoco testA|       |       | Refcoco testB| ||
-|-------------------------------------------------------------------------------------------------------|------|------|---------|------|-------|------|-----|------|------|
-| Model                                                                                                 | oIoU | mIoU | [email protected] | oIoU | mIoU  |[email protected] | oIoU | mIoU  |[email protected]  |
-| [PolyFormer-B](https://drive.google.com/file/d/1K0y-WBO6cL7gBzNnJaHAeNu3pgq4DbJ9/view?usp=share_link) | 74.82| 75.96 | 89.73   |76.64| 77.09 | 91.73| 71.06| 73.22 | 86.03 |
-| [PolyFormer-L](https://drive.google.com/file/d/15P6m5RI6HAQE2QXQXMAjw_oBsaPii7b3/view?usp=share_link) | 75.96| 76.94 | 90.38   |78.29| 78.49 | 92.89| 73.25| 74.83 |  87.16|
-|                                  [test_demo.py](..%2F..%2FDownloads%2Ftest_demo.py)                                                                     | Refcoco val|           | | Refcoco testA|       |       | Refcoco testB| ||
-|--------------------------------------------------------------------------------------------------------|------|------|------|------|------|------|------|------|------|
-| Model                                                                                                  | oIoU | mIoU |[email protected]| oIoU | mIoU  |[email protected] | oIoU | mIoU  |[email protected]  |
-| [PolyFormer-B ](https://drive.google.com/file/d/12_ylFhsbqGySxDqgeEByn8nKoJtT2n2w/view?usp=share_link) |  67.64| 70.65 | 83.73 | 72.89| 74.51 | 88.60 | 59.33| 64.64 | 76.38 | 67.76| 69.36  |
-| [PolyFormer-L](https://drive.google.com/file/d/1lUCv7dUPctEz4vEpPr7aI8A8ZmfYCB8y/view?usp=share_link)  |  69.33| 72.15 | 84.98 | 74.56| 75.71 | 89.77 | 61.87| 66.73 | 77.97 | 69.20| 71.15 |
-|                                                                                                       |  Refcocog val| || | Refcocog test| |
-|-------------------------------------------------------------------------------------------------------|------|------|------|------|------|------|
-| Model                                                                                                 |  oIoU | mIoU   |[email protected] | oIoU | mIoU   |[email protected] |
-| [PolyFormer-B](https://drive.google.com/file/d/12_ylFhsbqGySxDqgeEByn8nKoJtT2n2w/view?usp=share_link) |  67.76| 69.36  | 84.46| 69.05| 69.88 | 84.96 |
-| [PolyFormer-L](https://drive.google.com/file/d/1lUCv7dUPctEz4vEpPr7aI8A8ZmfYCB8y/view?usp=share_link) |  69.20| 71.15 | 85.83 | 70.19| 71.17  | 85.91|
-* Pretrained weights:
-  * [PolyFormer-B](https://drive.google.com/file/d/1sAzfChYDdHdaeatB2K14lrJjG4uiXAol/view?usp=share_link)
-  * [PolyFormer-L](https://drive.google.com/file/d/1knRxgM1lmEkuZZ-cOm_fmwKP1H0bJGU9/view?usp=share_link)
-# Acknowlegement
-This codebase is developed based on [OFA](https://github.com/OFA-Sys/OFA).
-Other related codebases include:
-* [Fairseq](https://github.com/pytorch/fairseq)
-* [refer](https://github.com/lichengunc/refer)
-* [LAVT-RIS](https://github.com/yz93/LAVT-RIS/)
-* [SeqTR](https://github.com/sean-zhuh/SeqTR)
-# Citation
-Please cite our paper if you find this codebase helpful :)
-```
-@inproceedings{liu2023polyformer,
-  title={PolyFormer: Referring Image Segmentation as Sequential Polygon Generation},
-  author={Liu, Jiang and Ding, Hui and Cai, Zhaowei and Zhang, Yuting and Satzoda, Ravi Kumar and Mahadevan, Vijay and Manmatha, R},
-  booktitle={CVPR},
-  year={2023}
-}
-```
-## Security
-See [CONTRIBUTING](CONTRIBUTING.md#security-issue-notifications) for more information.
-## License
-This project is licensed under the Apache-2.0 License.

+---
+title: PolyFormer
+emoji: 🖌️🎨
+colorFrom: pink
+colorTo: purple
+sdk: gradio
+sdk_version: 3.14.0
+app_file: app.py
+pinned: false
+license: afl-3.0
+---
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference