save

2023-03-04 12:36:00 +00:00
parent 165adf7953
commit 7c92e03b45
28 changed files with 1216 additions and 77 deletions
@@ -8,7 +8,7 @@ import numpy as np
 import plotly.graph_objects as go
 import plotly.express as ex

-os.environ["CUDA_VISIBLE_DEVICES"] = "6"
+os.environ["CUDA_VISIBLE_DEVICES"] = "4"

 gpus = tf.config.experimental.list_physical_devices('GPU')
 if gpus:
@@ -197,7 +197,7 @@ if __name__ == '__main__':
    sizes = [32, 64, 128, 256]
    folder = args.folder

-    is_skirt = True
+    is_skirt = False
    uv_masks_body = dict()
    uv_mask_body = np.load('custom_utils/mask_256x512' + ('_skirt' if is_skirt else '') + '.npy')
    uv_mask_body = cv2.dilate(uv_mask_body.astype(np.float32), cv2.getStructuringElement(cv2.MORPH_RECT, (2 * 2 + 1, 2 * 2 + 1), (2, 2)), iterations=2)