javirk / person_remover

People removal in images using Pix2Pix and YOLO.

Hey, I'm Javier 👋

I'm an AI researcher with very wide interests from Computer Vision to Simulation. I was the creator of Person Remover👻 and co-creator of Virtual Walk🏃‍♂️.

Now, I am working on rendering and doing my first steps in cell simulation while learning Rust🦀. In AI, I am interested in Neural Rendering.

Visit more of my work on my personal webpage: https://www.javiergamazo.com/

person_remover's Issues

Problem converting YOLO weights

Hi,

Im having issues converting the YOLO weights. After downloading the model Im getting this error.

python convert.py
Traceback (most recent call last):
File "convert.py", line 4, in <module>
    from yolo.yolov3.models import YoloV3, YoloV3Tiny
    ModuleNotFoundError: No module named 'yolo.yolov3'

Not able to generate outputs

I am able to run person_remover.py . It printing some arrays in console but isn't saving any figures to output dir. Pls suggest a soln.

Training the model

for training , in instructions why is mode set as test? in test mode checkpoints are not saved in pix2pix/checkpoint right?
and if I have my own data to train the model and save check point to later use in person_remover.py , which mode should I use train or test?

problem in convert.py

Hi,

when you are converting the yolov3.weights at that time you have given the name yolov3_class.tf but when you are loading, you have called the file as yolov3.tf . And, on first place there is no file getting saved with name tf extention. I got 3 files with ext .tf.index , yolov3_class.tf.data-00000-of-00002 ,yolov3_class.tf.data-00001-of-00002 which one shall i load?

Use in android

I want to develop an app in android. Can I use this tenserflow build ? if yes how to use it ???

Removing other objects

How can i remove other objects in scene like cars etc?

(Another) problem in convert.py

Hi,

When I try to run convert.py, I get:

I1103 08:21:42.501203 139996653496128 utils.py:45] yolo_output_2/conv2d_74 bias
Traceback (most recent call last):
  File "convert.py", line 34, in <module>
    app.run(main)
  File "/python3.6/site-packages/absl/app.py", line 300, in run
    _run_main(main, args)
  File "/python3.6/site-packages/absl/app.py", line 251, in _run_main
    sys.exit(main(argv))
  File "convert.py", line 21, in main
    load_darknet_weights(yolo, FLAGS.weights, FLAGS.tiny)
  File "/Person_remover/yolo/yolov3/utils.py", line 74, in load_darknet_weights
    assert len(wf.read()) == 0, 'failed to read all data'
AssertionError: failed to read all data

any ideas?