Code Monkey home page Code Monkey logo

cmasktrack-rcnn's People

Contributors

qjy981010 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar

cmasktrack-rcnn's Issues

Making intermediate frames available

Hi

I just discovered the OVIS dataset and it looks really useful to us. Great work. I realized that the frames are annotated at 5 fps and since I would like to use this dataset for training I am planning to interpolate the annotations such that I get a higher frame rate of for example 30 fps (6 times more training data). However, I cannot find all the frames. For example TAO makes all frames available but the annotations are only available on every 30th frame. Could you share the intermediate video frames (even without annotations)? It would also be interesting to analyze if running the methods on a higher frame rate helps to achieve a higher accuracy.

Thanks in advance and best,
Christoph

"RuntimeError: CUDA error: invalid device function" during training

Some problems occur when I launch the training script. The error information is as follow,
Traceback (most recent call last):
File "train.py", line 103, in
main()
File "train.py", line 99, in main
logger=logger)
File "/data/liangzhiyuan/projects/CMaskTrack/mmdet/apis/train.py", line 60, in train_detector
_non_dist_train(model, dataset, cfg, validate=validate, logger = logger)
File "/data/liangzhiyuan/projects/CMaskTrack/mmdet/apis/train.py", line 124, in _non_dist_train
runner.run(data_loaders, cfg.workflow, cfg.total_epochs)
File "/data/liangzhiyuan/projects/CMaskTrack/mmcv/runner/runner.py", line 358, in run
epoch_runner(data_loaders[i], **kwargs)
File "/data/liangzhiyuan/projects/CMaskTrack/mmcv/runner/runner.py", line 264, in train
self.model, data_batch, train_mode=True, **kwargs)
File "/data/liangzhiyuan/projects/CMaskTrack/mmdet/apis/train.py", line 38, in batch_processor
losses = model(**data)
File "/home/liangzhiyuan/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, **kwargs)
File "/home/liangzhiyuan/.local/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 152, in forward
outputs = self.parallel_apply(replicas, inputs, kwargs)
File "/home/liangzhiyuan/.local/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 162, in parallel_apply
return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])
File "/home/liangzhiyuan/.local/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 85, in parallel_apply
output.reraise()
File "/home/liangzhiyuan/.local/lib/python3.6/site-packages/torch/_utils.py", line 394, in reraise
raise self.exc_type(msg)
RuntimeError: Caught RuntimeError in replica 0 on device 0.
Original Traceback (most recent call last):
File "/home/liangzhiyuan/.local/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 60, in _worker
output = module(*input, **kwargs)
File "/home/liangzhiyuan/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call
result = self.forward(*input, **kwargs)
File "/data/liangzhiyuan/projects/CMaskTrack/mmdet/models/detectors/base.py", line 85, in forward
return self.forward_train(img, img_meta, **kwargs)
File "/data/liangzhiyuan/projects/CMaskTrack/mmdet/models/detectors/two_stage.py", line 117, in forward_train
proposal_list = self.rpn_head.get_bboxes(*proposal_inputs)
File "/data/liangzhiyuan/projects/CMaskTrack/mmdet/models/anchor_heads/anchor_head.py", line 232, in get_bboxes
scale_factor, cfg, rescale)
File "/data/liangzhiyuan/projects/CMaskTrack/mmdet/models/anchor_heads/rpn_head.py", line 80, in get_bboxes_single
proposals, _ = nms(proposals, cfg.nms_thr)
File "/data/liangzhiyuan/projects/CMaskTrack/mmdet/ops/nms/nms_wrapper.py", line 35, in nms
return dets[inds, :], inds
RuntimeError: CUDA error: invalid device function

The pytorch version is 1.3.1 with cuda10.1, and the torchvision is 0.2.2.

torchScript Export

Hi!

Would it be possible to add a TorchScript export file? This addition would be extremely helpful for utilizing it with various third-party software.

Thank you,

Paul

ModuleNotFoundError: No module named 'mmdet.ops.nms.gpu_nms'

Following install instructions. Running test_video.py returns:

File "CMaskTrack-RCNN/mmdet/core/post_processing/init.py", line 1, in
from .bbox_nms import multiclass_nms
File "CMaskTrack-RCNN/mmdet/core/post_processing/bbox_nms.py", line 3, in
from mmdet.ops.nms import nms_wrapper
File "CMaskTrack-RCNN/mmdet/ops/init.py", line 1, in
from .nms import nms, soft_nms
File "CMaskTrack-RCNN/mmdet/ops/nms/init.py", line 1, in
from .nms_wrapper import nms, soft_nms
File "CMaskTrack-RCNN/mmdet/ops/nms/nms_wrapper.py", line 5, in
from mmdet.ops.nms.gpu_nms import gpu_nms
ModuleNotFoundError: No module named 'mmdet.ops.nms.gpu_nms'

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.