Code Monkey home page Code Monkey logo

yoeo's People

Contributors

0asa avatar cahity avatar catalinolaru1 avatar confusedlama avatar ctrlxx avatar dependabot[bot] avatar developer0hye avatar eriklindernoren avatar everwinter23 avatar flova avatar hnishi avatar id9502 avatar jaagut avatar johagge avatar joshuachough avatar phinik avatar timforby avatar timonegk avatar triwahyuu avatar v-iashin avatar wptoux avatar yrrah avatar

Stargazers

 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  avatar  avatar  avatar  avatar  avatar

yoeo's Issues

Dataloader should not convert segmentation images

Context

As discussed in #61 (#61 (comment)), the YOEO dataloader should not convert segmentation images from TORSO-21 format (0, 127, 255, ...) to YOEO format (0,1,2, ...) itself.
This is the task of the TORSO-21 convert-script.
Instead, we should use the correct path yoeo_segmentations.

Directory name collision between YOEO and TORSO-21

  • YOEO expects segmentation mask images inside a directory called segmentations (see here).
    • Mask are machine-seeable: very dark colors (0, 1, 2, ...)
    • createYoeoLabelsTORSO.py script (before #60) outputs them at yoeo_masks directory (see here).
  • TORSO-21 provides segmentation mask images inside a directory called segmentaions
    • Masks are human-seeable: colors are distinct (0, 127, 255, ...)

Expected behavior

The directory names should not collide, i.e., should not be identical.

Current behavior

Converted masks are in yoeo_masks directory.
User has to rename directory and resolve name conflict manually.

Possible Solution

  • YOEO's dataloader reads from a differently called directory like yoeo_masks
  • ...

CI features

Here are some ideas what to do in our CI

  • Test for determinism (training produces the same output?)
  • Test conversion scripts

Originally posted by @jaagut in #44 (comment)

Update `create_custom_model` script

The output of the create_custom_model highly differs from the current yoeo.cfg.
The script is still identical to the original PyTorchYolo script.
We should update this. In the same breath, we could convert this bash script to a python script and create a custom command (update Readme!)

Return class index as uint8 not int64

It would be nice to return the class index of the segmentation as an unit8 for performance reasons. This should lower USB copy times from NCS2 and improve overall performance, as well as simplify the code in the bitbots_vision.

Proposed change

Add an uint8 typecast to the following line.

return torch.argmax(x, dim=1)

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.