Coder Social home page Coder Social logo

Training on custom dataset about oneformer HOT 3 CLOSED

shi-labs avatar shi-labs commented on May 20, 2024
Training on custom dataset

from oneformer.

Comments (3)

praeclarumjj3 avatar praeclarumjj3 commented on May 20, 2024

Hi @sushilkhadkaanon, thanks for your interest in our work.

I believe the instructions are comprehensive enough, as other users (#17, #31) have been able to train on custom datasets following my instructions. I suggest you spend some more time trying to understand registering custom datasets in Detectron2 and then try preparing a custom dataset mapper.

Let me know if you have any specific issues.

from oneformer.

sushilkhadkaanon avatar sushilkhadkaanon commented on May 20, 2024

Hi! Thanks, I really appreciate your work. I tried following your instructions.
I added InstanceCOCOCustomNewBaselineDatasetMapper in build_train_loader of train_net.py
But where do I register my dataset?
` #

from detectron2.data.datasets import register_coco_instances

register_coco_instances("my_dataset_train", {}, "_annotations.coco.json", "custom_data/train")
register_coco_instances("my_dataset_test", {}, "_annotations.coco.json", "custom_data/train")

metadata_train = MetadataCatalog.get("my_dataset_train")
cfg.DATASETS.TRAIN = ("my_dataset_train",)
cfg.DATASETS.TEST = ("my_dataset_test",)

cfg.SOLVER.MAX_ITER = 5 

` 

Added this inside main function in train_net.py.
Now this is giving me `ModuleNotFoundError: 

  Please compile MultiScaleDeformableAttention CUDA op with the following commands:
\`cd mask2former/modeling/pixel_decoder/ops\`
\`sh make.sh\``

from oneformer.

praeclarumjj3 avatar praeclarumjj3 commented on May 20, 2024

You can look at register_ade20k_instance.py for reference. You need to create a similar file for your dataset. Also, the official detectron2 guide on adding custom datasets is quite detailed.

from oneformer.

Related Issues (20)

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.