search

HypoX64 / DeepMosaics

Automatically remove the mosaics in images and videos, or add mosaics to them.
GNU General Public License v3.0
2.08k stars 433 forks source link

how can i use multi processing? #65

Open skyzombie89 opened 3 years ago

skyzombie89 commented 3 years ago

do i have to put command to use multi - processing? or it just automatically works?

ethanfel commented 3 years ago

With gpu ? export CUDA_DEVICE_ORDER=PCI_BUS_ID parameter : --gpu_id 0,1,2,3,..

Engrish commented 2 years ago

Tried to add that to Windows env but got the below error:

DeepMosaics: 0.5.1 Python: 3.6.2 |Anaconda, Inc.| (default, Sep 30 2017, 11:52:29) [MSC v.1900 64 bit (AMD64)] Pytorch: 1.8.0a0+37c1f4a OpenCV: 4.5.1 Platform: Windows-10-10.0.19041-SP0 --------------BUG-------------- Error Type: <class 'RuntimeError'> Error(s) in loading state_dict for DataParallel: Missing key(s) in state_dict: "module.encoder3d.model.0.weight_orig", "module.encoder3d.model.0.weight", "module.encoder3d.model.0.weight_u", "module.encoder3d.model.0.bias", "module.encoder3d.model.0.weight_orig", "module.encoder3d.model.0.weight_u", "module.encoder3d.model.0.weight_v", "module.encoder3d.model.2.weight_orig", "module.encoder3d.model.2.weight", "module.encoder3d.model.2.weight_u", "module.encoder3d.model.2.bias", "module.encoder3d.model.2.weight_orig", "module.encoder3d.model.2.weight_u", "module.encoder3d.model.2.weight_v", "module.encoder3d.model.4.weight_orig", "module.encoder3d.model.4.weight", "module.encoder3d.model.4.weight_u", "module.encoder3d.model.4.bias", "module.encoder3d.model.4.weight_orig", "module.encoder3d.model.4.weight_u", "module.encoder3d.model.4.weight_v", "module.encoder3d.model.6.weight_orig", "module.encoder3d.model.6.weight", "module.encoder3d.model.6.weight_u", "module.encoder3d.model.6.bias", "module.encoder3d.model.6.weight_orig", "module.encoder3d.model.6.weight_u", "module.encoder3d.model.6.weight_v", "module.encoder2d.model.1.weight_orig", "module.encoder2d.model.1.weight", "module.encoder2d.model.1.weight_u", "module.encoder2d.model.1.bias", "module.encoder2d.model.1.weight_orig", "module.encoder2d.model.1.weight_u", "module.encoder2d.model.1.weight_v", "module.encoder2d.model.4.weight_orig", "module.encoder2d.model.4.weight", "module.encoder2d.model.4.weight_u", "module.encoder2d.model.4.bias", "module.encoder2d.model.4.weight_orig", "module.encoder2d.model.4.weight_u", "module.encoder2d.model.4.weight_v", "module.encoder2d.model.7.weight_orig", "module.encoder2d.model.7.weight", "module.encoder2d.model.7.weight_u", "module.encoder2d.model.7.bias", "module.encoder2d.model.7.weight_orig", "module.encoder2d.model.7.weight_u", "module.encoder2d.model.7.weight_v", "module.encoder2d.model.10.weight_orig", "module.encoder2d.model.10.weight", "module.encoder2d.model.10.weight_u", "module.encoder2d.model.10.bias", "module.encoder2d.model.10.weight_orig", "module.encoder2d.model.10.weight_u", "module.encoder2d.model.10.weight_v", "module.blocks.0.conv_block.1.weight_orig", "module.blocks.0.conv_block.1.weight", "module.blocks.0.conv_block.1.weight_u", "module.blocks.0.conv_block.1.bias", "module.blocks.0.conv_block.1.weight_orig", "module.blocks.0.conv_block.1.weight_u", "module.blocks.0.conv_block.1.weight_v", "module.blocks.0.conv_block.4.weight_orig", "module.blocks.0.conv_block.4.weight", "module.blocks.0.conv_block.4.weight_u", "module.blocks.0.conv_block.4.bias", "module.blocks.0.conv_block.4.weight_orig", "module.blocks.0.conv_block.4.weight_u", "module.blocks.0.conv_block.4.weight_v", "module.blocks.1.conv_block.1.weight_orig", "module.blocks.1.conv_block.1.weight", "module.blocks.1.conv_block.1.weight_u", "module.blocks.1.conv_block.1.bias", "module.blocks.1.conv_block.1.weight_orig", "module.blocks.1.conv_block.1.weight_u", "module.blocks.1.conv_block.1.weight_v", "module.blocks.1.conv_block.4.weight_orig", "module.blocks.1.conv_block.4.weight", "module.blocks.1.conv_block.4.weight_u", "module.blocks.1.conv_block.4.bias", "module.blocks.1.conv_block.4.weight_orig", "module.blocks.1.conv_block.4.weight_u", "module.blocks.1.conv_block.4.weight_v", "module.blocks.2.conv_block.1.weight_orig", "module.blocks.2.conv_block.1.weight", "module.blocks.2.conv_block.1.weight_u", "module.blocks.2.conv_block.1.bias", "module.blocks.2.conv_block.1.weight_orig", "module.blocks.2.conv_block.1.weight_u", "module.blocks.2.conv_block.1.weight_v", "module.blocks.2.conv_block.4.weight_orig", "module.blocks.2.conv_block.4.weight", "module.blocks.2.conv_block.4.weight_u", "module.blocks.2.conv_block.4.bias", "module.blocks.2.conv_block.4.weight_orig", "module.blocks.2.conv_block.4.weight_u", "module.blocks.2.conv_block.4.weight_v", "module.blocks.3.conv_block.1.weight_orig", "module.blocks.3.conv_block.1.weight", "module.blocks.3.conv_block.1.weight_u", "module.blocks.3.conv_block.1.bias", "module.blocks.3.conv_block.1.weight_orig", "module.blocks.3.conv_block.1.weight_u", "module.blocks.3.conv_block.1.weight_v", "module.blocks.3.conv_block.4.weight_orig", "module.blocks.3.conv_block.4.weight", "module.blocks.3.conv_block.4.weight_u", "module.blocks.3.conv_block.4.bias", "module.blocks.3.conv_block.4.weight_orig", "module.blocks.3.conv_block.4.weight_u", "module.blocks.3.conv_block.4.weight_v", "module.decoder.0.convup.2.weight_orig", "module.decoder.0.convup.2.weight", "module.decoder.0.convup.2.weight_u", "module.decoder.0.convup.2.bias", "module.decoder.0.convup.2.weight_orig", "module.decoder.0.convup.2.weight_u", "module.decoder.0.convup.2.weight_v", "module.decoder.1.convup.2.weight_orig", "module.decoder.1.convup.2.weight", "module.decoder.1.convup.2.weight_u", "module.decoder.1.convup.2.bias", "module.decoder.1.convup.2.weight_orig", "module.decoder.1.convup.2.weight_u", "module.decoder.1.convup.2.weight_v", "module.decoder.2.convup.2.weight_orig", "module.decoder.2.convup.2.weight", "module.decoder.2.convup.2.weight_u", "module.decoder.2.convup.2.bias", "module.decoder.2.convup.2.weight_orig", "module.decoder.2.convup.2.weight_u", "module.decoder.2.convup.2.weight_v", "module.decoder.4.weight", "module.decoder.4.bias". Unexpected key(s) in state_dict: "encoder3d.model.0.bias", "encoder3d.model.0.weight_orig", "encoder3d.model.0.weight_u", "encoder3d.model.0.weight_v", "encoder3d.model.2.bias", "encoder3d.model.2.weight_orig", "encoder3d.model.2.weight_u", "encoder3d.model.2.weight_v", "encoder3d.model.4.bias", "encoder3d.model.4.weight_orig", "encoder3d.model.4.weight_u", "encoder3d.model.4.weight_v", "encoder3d.model.6.bias", "encoder3d.model.6.weight_orig", "encoder3d.model.6.weight_u", "encoder3d.model.6.weight_v", "encoder2d.model.1.bias", "encoder2d.model.1.weight_orig", "encoder2d.model.1.weight_u", "encoder2d.model.1.weight_v", "encoder2d.model.4.bias", "encoder2d.model.4.weight_orig", "encoder2d.model.4.weight_u", "encoder2d.model.4.weight_v", "encoder2d.model.7.bias", "encoder2d.model.7.weight_orig", "encoder2d.model.7.weight_u", "encoder2d.model.7.weight_v", "encoder2d.model.10.bias", "encoder2d.model.10.weight_orig", "encoder2d.model.10.weight_u", "encoder2d.model.10.weight_v", "blocks.0.conv_block.1.bias", "blocks.0.conv_block.1.weight_orig", "blocks.0.conv_block.1.weight_u", "blocks.0.conv_block.1.weight_v", "blocks.0.conv_block.4.bias", "blocks.0.conv_block.4.weight_orig", "blocks.0.conv_block.4.weight_u", "blocks.0.conv_block.4.weight_v", "blocks.1.conv_block.1.bias", "blocks.1.conv_block.1.weight_orig", "blocks.1.conv_block.1.weight_u", "blocks.1.conv_block.1.weight_v", "blocks.1.conv_block.4.bias", "blocks.1.conv_block.4.weight_orig", "blocks.1.conv_block.4.weight_u", "blocks.1.conv_block.4.weight_v", "blocks.2.conv_block.1.bias", "blocks.2.conv_block.1.weight_orig", "blocks.2.conv_block.1.weight_u", "blocks.2.conv_block.1.weight_v", "blocks.2.conv_block.4.bias", "blocks.2.conv_block.4.weight_orig", "blocks.2.conv_block.4.weight_u", "blocks.2.conv_block.4.weight_v", "blocks.3.conv_block.1.bias", "blocks.3.conv_block.1.weight_orig", "blocks.3.conv_block.1.weight_u", "blocks.3.conv_block.1.weight_v", "blocks.3.conv_block.4.bias", "blocks.3.conv_block.4.weight_orig", "blocks.3.conv_block.4.weight_u", "blocks.3.conv_block.4.weight_v", "decoder.0.convup.2.bias", "decoder.0.convup.2.weight_orig", "decoder.0.convup.2.weight_u", "decoder.0.convup.2.weight_v", "decoder.1.convup.2.bias", "decoder.1.convup.2.weight_orig", "decoder.1.convup.2.weight_u", "decoder.1.convup.2.weight_v", "decoder.2.convup.2.bias", "decoder.2.convup.2.weight_orig", "decoder.2.convup.2.weight_u", "decoder.2.convup.2.weight_v", "decoder.4.weight", "decoder.4.bias". <FrameSummary file deepmosaic.py, line 77 in <module>> <FrameSummary file deepmosaic.py, line 41 in main> <FrameSummary file c:\deepm\DeepMosaics\models\loadmodel.py, line 56 in video> <FrameSummary file c:\condaEnv3\lib\site-packages\torch\nn\modules\module.py, line 1224 in load_state_dict>