search

neuralchen / SimSwap

An arbitrary face-swapping framework on images and videos with one single trained model!
Other
4.52k stars 891 forks source link

Google Colab notebook returns (AttributeError: 'NoneType' object has no attribute 'shape') #149

Closed neptune-gaming closed 2 years ago

neptune-gaming commented 2 years ago

------------ Options ------------- Arc_path: models/BEST_checkpoint.tar aspect_ratio: 1.0 batchSize: 8 checkpoints_dir: ./checkpoints cluster_path: features_clustered_010.npy data_type: 32 dataroot: ./datasets/cityscapes/ display_winsize: 512 engine: None export_onnx: None f: /root/.local/share/jupyter/runtime/kernel-67c2d95c-f867-481f-8e48-3fafd4099b20.json feat_num: 3 fineSize: 512 fp16: False gpu_ids: [0] how_many: 50 id_thres: 0.03 image_size: 224 input_nc: 3 instance_feat: False isTrain: False label_feat: False label_nc: 0 latent_size: 512 loadSize: 1024 load_features: False local_rank: 0 max_dataset_size: inf model: pix2pixHD multisepcific_dir: ./demo_file/multispecific nThreads: 2 n_blocks_global: 6 n_blocks_local: 3 n_clusters: 10 n_downsample_E: 4 n_downsample_global: 3 n_local_enhancers: 1 name: people nef: 16 netG: global ngf: 64 niter_fix_global: 0 no_flip: False no_instance: False no_simswaplogo: False norm: batch norm_G: spectralspadesyncbatch3x3 ntest: inf onnx: None output_nc: 3 output_path: ./output/ phase: test pic_a_path: ./crop_224/gdg.jpg pic_b_path: ./crop_224/zrf.jpg pic_specific_path: ./crop_224/zrf.jpg resize_or_crop: scale_width results_dir: ./results/ semantic_nc: 3 serial_batches: False temp_path: ./temp_results tf_log: False use_dropout: False use_encoded_image: False use_mask: False verbose: False video_path: ./demo_file/multi_people_1080p.mp4 which_epoch: latest -------------- End ---------------- input mean and std: 127.5 127.5 find model: ./insightface_func/models/antelope/glintr100.onnx recognition find model: ./insightface_func/models/antelope/scrfd_10g_bnkps.onnx detection set det-size: (640, 640)

AttributeError Traceback (most recent call last)

in () 24 # img_a = Image.open(pic_a).convert('RGB') 25 img_a_whole = cv2.imread(pic_a) ---> 26 img_a_align_crop, _ = app.get(img_a_whole,crop_size) 27 img_a_align_crop_pil = Image.fromarray(cv2.cvtColor(img_a_align_crop[0],cv2.COLOR_BGR2RGB)) 28 img_a = transformer_Arcface(img_a_align_crop_pil) 1 frames /usr/local/lib/python3.7/dist-packages/insightface/model_zoo/scrfd.py in detect(self, img, threshold, input_size, max_num, metric) 202 input_size = self.input_size if input_size is None else input_size 203 --> 204 im_ratio = float(img.shape[0]) / img.shape[1] 205 model_ratio = float(input_size[1]) / input_size[0] 206 if im_ratio>model_ratio: AttributeError: 'NoneType' object has no attribute 'shape'
neptune-gaming commented 2 years ago

Nvm. I found out it was an issue with my picture.