quereste / implicit-deepfake Goto Github PK

Hello, great notebook,
but I have a problem with the following cell:

'''deepfake_source_img_path = '/content/drive/MyDrive/MyImplicitDeepfake/johan_00003_.jpg' # @param {type:"string"}
dirs_dict = {'train': 100, 'val': 50, 'test': 10}
batch_size = 40 # @param {type:"integer"}

create_deepfakes(dirs_dict, base_dir, deepfake_source_img_path, batch_size)
convert_to_rgba()'''

Give me this error:
Everything is ok!

AttributeError Traceback (most recent call last)
in <cell line: 5>()
3 batch_size = 40 # @param {type:"integer"}
4
----> 5 create_deepfakes(dirs_dict, base_dir, deepfake_source_img_path, batch_size)
6 convert_to_rgba()

4 frames
/usr/local/lib/python3.10/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'

Could you help me fix this issue

Hi, first of all very interesting work!
I wanted to try it on a custom 3d head model.
Which tool did you use to get the camera parameters? I tried with BlenderNeRF but the images are from not as good angles as in the colab dataset provided.