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

Refine DensePose codes and fix minor bugs #215

Open
wants to merge 6 commits into
base: main
Choose a base branch
from

Conversation

Johnqczhang
Copy link

Hi, thanks for sharing the code of such a good work. This PR mainly refines codes related to Body UV head network by formatting the code in the canonical style and fixes some minor bugs as shown in related commits. An additional function of visualizing training metrics in a tensorboard is added which is compatible with current framework and easy to be enabled by a simple command-line argument.

Major modifications include:

  1. Fix a typo in config.py;

  2. Refine code style in vis.py and add more comments for understanding more easily;

  3. Refine densepose evaluation in densepose_cocoeval.py by simplifying the extraction of geodesic distance between two vertices from the pre-computed matrix in a compact representaion. Code style is also refined;

  4. Fix broken links in INSTALL.md and README.md (similar PR in Fix broken link to detectron in Install.md #211);

  5. Refine the implementation of building body_uv_rcnn head network before the real training process starts. Specifically,

  1. Refine loading a file of RoI proposals or model weights in io.py and add an additional function of visualizing training metrics (classification accuracy and losses) during the training process by using tensorboardX without hurting the compatibility of the whole framework.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
CLA Signed Do not delete this pull request or issue due to inactivity.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants