maskrcnn_benchmark/modeling/rpn/retinanet/inference.py (4 lines): - line 95: # TODO most of this can be made out of the loop for - line 97: # TODO:Yang: Not easy to do. Because the numbers of detections are - line 126: # TODO very similar to filter_results from PostProcessor - line 128: # TODO Yang: solve this issue in the future. No good solution maskrcnn_benchmark/structures/keypoint.py (4 lines): - line 10: # FIXME remove check once we have better integration with device - line 18: # TODO should I split them? - line 129: # TODO this doesn't look great - line 153: # TODO make this nicer, this is a direct translation from C2 (but removing the inner loop) maskrcnn_benchmark/csrc/cuda/ROIPool_cuda.cu (3 lines): - line 10: // TODO make it in a common file - line 155: // TODO remove the dependency on input and use instead its sizes -> save memory - line 169: // TODO add more checks maskrcnn_benchmark/modeling/roi_heads/keypoint_head/inference.py (2 lines): - line 35: # TODO remove and use only the Keypointer - line 111: # TODO do this properly maskrcnn_benchmark/csrc/cuda/ROIAlign_cuda.cu (2 lines): - line 9: // TODO make it in a common file - line 301: // TODO remove the dependency on input and use instead its sizes -> save memory maskrcnn_benchmark/data/datasets/evaluation/cityscapes/eval_instances.py (2 lines): - line 225: # TODO: not parallel! parallelize this process safely - line 378: # TODO: Issue #527 - bad Masker interface maskrcnn_benchmark/csrc/cuda/nms.cu (2 lines): - line 83: THCState *state = at::globalContext().lazyInitCUDA(); // TODO replace with getTHCState - line 126: // TODO improve this part maskrcnn_benchmark/modeling/roi_heads/mask_head/inference.py (2 lines): - line 10: # TODO check if want to return a single BoxList or a composite - line 192: # TODO: Is this JIT compatible? maskrcnn_benchmark/structures/boxlist_ops.py (2 lines): - line 42: # TODO maybe add an API for querying the ws / hs - line 92: # TODO redundant, remove maskrcnn_benchmark/modeling/box_coder.py (2 lines): - line 32: TO_REMOVE = 1 # TODO remove - line 64: TO_REMOVE = 1 # TODO remove maskrcnn_benchmark/data/datasets/evaluation/coco/coco_eval.py (2 lines): - line 166: # TODO replace with get_img_info? - line 232: # TODO maybe remove this and make it explicit in the documentation maskrcnn_benchmark/modeling/backbone/resnet.py (1 line): - line 285: # TODO: specify init for the above tools/train_net.py (1 line): - line 103: torch.cuda.empty_cache() # TODO check if it helps maskrcnn_benchmark/layers/sigmoid_focal_loss.py (1 line): - line 8: # TODO: Use JIT to replace CUDA implementation in the future. maskrcnn_benchmark/solver/lr_scheduler.py (1 line): - line 7: # FIXME ideally this would be achieved with a CombinedLRScheduler, maskrcnn_benchmark/layers/smooth_l1_loss.py (1 line): - line 5: # TODO maybe push this to nn? maskrcnn_benchmark/data/datasets/coco.py (1 line): - line 72: # TODO might be better to add an extra field maskrcnn_benchmark/structures/image_list.py (1 line): - line 52: # TODO Ideally, just remove this and let me model handle arbitrary maskrcnn_benchmark/csrc/nms.h (1 line): - line 16: // TODO raise error if not compiled with CUDA maskrcnn_benchmark/structures/bounding_box.py (1 line): - line 180: # TODO should I filter empty boxes here? maskrcnn_benchmark/modeling/roi_heads/roi_heads.py (1 line): - line 25: # TODO rename x to roi_box_features, if it doesn't increase memory consumption maskrcnn_benchmark/data/samplers/distributed.py (1 line): - line 3: # FIXME remove this once c10d fixes the bug it has maskrcnn_benchmark/modeling/roi_heads/box_head/inference.py (1 line): - line 60: # TODO think about a representation of batch of boxes maskrcnn_benchmark/modeling/backbone/fpn.py (1 line): - line 61: # TODO use size instead of scale to make it robust to different sizes maskrcnn_benchmark/csrc/cuda/SigmoidFocalLoss_cuda.cu (1 line): - line 14: // TODO make it in a common file maskrcnn_benchmark/modeling/roi_heads/keypoint_head/loss.py (1 line): - line 93: # TODO check if this is the right one, as BELOW_THRESHOLD maskrcnn_benchmark/data/datasets/evaluation/coco/abs_to_coco.py (1 line): - line 190: # TODO: parallelize maskrcnn_benchmark/modeling/roi_heads/mask_head/loss.py (1 line): - line 31: # FIXME: CPU computation bottleneck, this should be parallelized maskrcnn_benchmark/modeling/rpn/retinanet/retinanet.py (1 line): - line 25: # TODO: Implement the sigmoid version first. maskrcnn_benchmark/data/datasets/evaluation/voc/voc_eval.py (1 line): - line 13: # TODO need to make the use_07_metric format available