Git Product home page Git Product logo

parts2words's Introduction

Parts2Words: Learning Joint Embedding of Point Clouds and Texts by Bidirectional Matching between Parts and Words

Introduction

This is the source code of Parts2Words: Learning Joint Embedding of Point Clouds and Texts by Bidirectional Matching between Parts and Words, an approch for 3D Shape-Text Matching based on local based matching method.

Requirements

pip install -r requirements.txt

Download data

We pack the shapenet data with segmentation annotation in h5 file, you can download the h5 file from here

data
|-- models
|   `-- shapenet
|       
`-- shapenet
    |-- shapenetv2_level_1.pkl
    |-- split_shape_captioner
    |   |-- test_03001627.txt
    |   |-- test_04379243.txt
    |   |-- train_03001627.txt
    |   `-- train_04379243.txt
    `-- vocab
        `-- shapenet.json

Set configure file

set data path in config/parts2words_default.yaml.

Train model

python train.py --config conig/parts2words_default.yaml

evaluate model

python val.py --config conig/parts2words_default.yaml

Reference

If you found this code useful, please cite the following paper:

@article{tang2021parts2words,
  title={parts2words: Learning Joint Embedding of Point Clouds and Text by Matching Parts to Words},
  author={Tang, Chuan and Yang, Xi and Wu, Bojian and Han, Zhizhong and Chang, Yi},
  journal={arXiv preprint arXiv:2107.01872},
  year={2021}
}

parts2words's People

Contributors

jlutangchuan avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar

parts2words's Issues

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.