search

limuhit / 360-Image-Compression

11 stars 1 forks source link

Could you share the training code? Thank you very much ! #1

Closed zsevenj closed 2 years ago

limuhit commented 2 years ago

Schedule:

  1. Train baseline model without entropy by trainDDP_IMP.py
  2. Train the whole model with the entropy by trainDDP_IMP_ENT.py
  3. Extract the importance map by extract_codes.py
  4. Train a post entropy model for the importance map by trainDDP_POST_ENT.py

Dataset: https://github.com/limuhit/pseudocylindrical_convolution