Skip to content
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

Commit

Permalink
WorkAround for `ValueError: No ground-truth boxes available for one o…
Browse files Browse the repository at this point in the history
…f the images during training` (#836)

* Update setup.py

* add a FORCE_CUDA flag

Following discussion [here](#167), this seemed the best solution

* Update Dockerfile

* Update setup.py

* add FORCE_CUDA as an ARG

* 	modified:   docker/Dockerfile
	modified:   setup.py

* small fix to readme of demo

* remove test print

* keep ARG_CUDA

* remove env value and use the one from ARG

* keep same formatting as source

* change proposed by @miguelvr

* Update INSTALL.md

* Update trainer.py

* Update trainer.py

* Update trainer.py

* Update trainer.py

* Update trainer.py

* Update setup.py
  • Loading branch information
obendidi authored and fmassa committed May 29, 2019
1 parent 4c6cd1a commit d269847
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions maskrcnn_benchmark/engine/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,10 @@ def do_train(
start_training_time = time.time()
end = time.time()
for iteration, (images, targets, _) in enumerate(data_loader, start_iter):

if any(len(target) < 1 for target in targets):
logger.error(f"Iteration={iteration + 1} || Image Ids used for training {_} || targets Length={[len(target) for target in targets]}" )
continue
data_time = time.time() - end
iteration = iteration + 1
arguments["iteration"] = iteration
Expand Down

0 comments on commit d269847

Please sign in to comment.