update

Red-Fairy · Feb 3, 2024 · 38a753d · 38a753d
1 parent 2236780
commit 38a753d
Show file tree

Hide file tree

Showing 9 changed files with 10 additions and 8 deletions.
diff --git a/models/__pycache__/base_model.cpython-311.pyc b/models/__pycache__/base_model.cpython-311.pyc
diff --git a/models/__pycache__/uocf_model.cpython-311.pyc b/models/__pycache__/uocf_model.cpython-311.pyc
diff --git a/models/base_model.py b/models/base_model.py
@@ -201,6 +201,8 @@ def load_module(module_name, load_method='unload'):
                 if self.opt.one2four:
                     if 'fg_position' in state_dict:
                         del state_dict['fg_position']
+                    if 'slots_init_fg' in state_dict:
+                        del state_dict['slots_init_fg']
                 incompatible = net.load_state_dict(state_dict, strict=False)
                 if incompatible.missing_keys and not self.opt.continue_train: # if continue train, ignore missing keys
                     for key in incompatible.missing_keys:

diff --git a/scripts/kitchen-matte/train/train-stage2-noplane.sh b/scripts/kitchen-matte/train/train-stage2-noplane.sh
@@ -7,7 +7,7 @@ python train.py --dataroot $DATAROOT --n_scenes 735 --n_img_each_scene 2 \
     --load_size 128 --n_samp 64 --input_size 128 --supervision_size 64 --frustum_size 64 \
     --model 'uocf' --lr 0.00015 \
     --seed 2024 \
-    --num_slots 2 --attn_iter 6 --nss_scale 7 --fg_object_size 3 \
+    --num_slots 5 --attn_iter 6 --nss_scale 7 --fg_object_size 3 \
     --stratified --fixed_locality \
     --bg_density_loss --depth_supervision --remove_duplicate --remove_duplicate_in 100 \
     --load_pretrain --load_pretrain_path './checkpoints/room-texture/stage1-1obj-noplane' \

diff --git a/scripts/kitchen-matte/train/train-stage2-plane.sh b/scripts/kitchen-matte/train/train-stage2-plane.sh
@@ -7,7 +7,7 @@ python train.py --dataroot $DATAROOT --n_scenes 735 --n_img_each_scene 2 \
     --load_size 128 --n_samp 64 --input_size 128 --supervision_size 64 --frustum_size 64 \
     --model 'uocf' --lr 0.00015 \
     --seed 2024 \
-    --num_slots 2 --attn_iter 6 --nss_scale 7 --fg_object_size 3 \
+    --num_slots 5 --attn_iter 6 --nss_scale 7 --fg_object_size 3 \
     --stratified --fixed_locality \
     --bg_density_loss --depth_supervision --remove_duplicate --remove_duplicate_in 100 \
     --load_pretrain --load_pretrain_path './checkpoints/room-texture/stage1-1obj-plane' \

diff --git a/scripts/kitchen-shiny/train/train-stage2-noplane.sh b/scripts/kitchen-shiny/train/train-stage2-noplane.sh
@@ -7,7 +7,7 @@ python train.py --dataroot $DATAROOT --n_scenes 324 --n_img_each_scene 2 \
     --load_size 128 --n_samp 64 --input_size 128 --supervision_size 64 --frustum_size 64 \
     --model 'uocf' --lr 0.00015 \
     --seed 2024 \
-    --num_slots 2 --attn_iter 6 --nss_scale 7 --fg_object_size 3 \
+    --num_slots 5 --attn_iter 6 --nss_scale 7 --fg_object_size 3 \
     --stratified --fixed_locality \
     --bg_density_loss --depth_supervision --remove_duplicate --remove_duplicate_in 100 \
     --load_pretrain --load_pretrain_path './checkpoints/room-texture/stage1-1obj-noplane' \

diff --git a/scripts/kitchen-shiny/train/train-stage2-plane.sh b/scripts/kitchen-shiny/train/train-stage2-plane.sh
@@ -7,7 +7,7 @@ python train.py --dataroot $DATAROOT --n_scenes 324 --n_img_each_scene 2 \
     --load_size 128 --n_samp 64 --input_size 128 --supervision_size 64 --frustum_size 64 \
     --model 'uocf' --lr 0.00015 \
     --seed 2024 \
-    --num_slots 2 --attn_iter 6 --nss_scale 7 --fg_object_size 3 \
+    --num_slots 5 --attn_iter 6 --nss_scale 7 --fg_object_size 3 \
     --stratified --fixed_locality \
     --bg_density_loss --depth_supervision --remove_duplicate --remove_duplicate_in 100 \
     --load_pretrain --load_pretrain_path './checkpoints/room-texture/stage1-1obj-plane' \

diff --git a/scripts/room-texture/train/train-stage2-noplane.sh b/scripts/room-texture/train/train-stage2-noplane.sh
@@ -7,10 +7,10 @@ python train.py --dataroot $DATAROOT --n_scenes 5000 --n_img_each_scene 2 \
     --load_size 128 --n_samp 64 --input_size 128 --supervision_size 64 --frustum_size 64 \
     --model 'uocf' --lr 0.00015 \
     --seed 2024 \
-    --num_slots 2 --attn_iter 6 --nss_scale 7 --fg_object_size 3 \
+    --num_slots 5 --attn_iter 6 --nss_scale 7 --fg_object_size 3 \
     --stratified --fixed_locality \
     --bg_density_loss --depth_supervision --remove_duplicate \
-    --load_pretrain --load_pretrain_path './stage1-1obj-noplane' \
+    --load_pretrain --load_pretrain_path './checkpoints/room-texture/stage1-1obj-noplane' \
     --load_encoder 'load_train' --load_slotattention 'load_train' --load_decoder 'load_train' --load_epoch 100 --one2four \
     --coarse_epoch 60 --niter 120 --percept_in 10 --no_locality_epoch 20 --dense_sample_epoch 20 \
     --scaled_depth --depth_scale_pred \

diff --git a/scripts/room-texture/train/train-stage2-plane.sh b/scripts/room-texture/train/train-stage2-plane.sh
@@ -5,9 +5,9 @@ python train.py --dataroot $DATAROOT --n_scenes 5000 --n_img_each_scene 2 \
     --checkpoints_dir 'checkpoints' --name 'room-texture' \
     --display_port $PORT --display_ncols 4 --print_freq 50 --display_freq 50 --save_epoch_freq 5 \
     --load_size 128 --n_samp 64 --input_size 128 --supervision_size 64 --frustum_size 64 \
-    --model 'uocf' --lr 0.00015 \
+    --model 'uocf' --lr 0.0003 \
     --seed 2024 \
-    --num_slots 2 --attn_iter 6 --nss_scale 7 --fg_object_size 3 \
+    --num_slots 5 --attn_iter 6 --nss_scale 7 --fg_object_size 3 \
     --stratified --fixed_locality \
     --bg_density_loss --depth_supervision --remove_duplicate \
     --load_pretrain --load_pretrain_path './checkpoints/room-texture/stage1-1obj-plane' \