search

jrieke / traingenerator

🧙 A web app to generate template code for machine learning
https://traingenerator.jrieke.com
MIT License
1.36k stars 181 forks source link

New Template: Semantic Segmentation; Generative Adversarial Networks; Image Caption; Style Transfer; #15

Open The-ML-Hero opened 3 years ago

The-ML-Hero commented 3 years ago

@jrieke

Hi, everybody, I'm looking forward to contributing this project can I add

  1. Semantic Segmentation
  2. Generative Adversarial Networks
  3. Image Caption
  4. Style Transfer

as I think this would greatly help beginners who are starting out their deep learning journey and want to do interesting things like these.

PS: can I also work on object detection using the YOLO-v5 achritexture.

Thanks a lot please reply as soon as possible

themrcesi commented 3 years ago

Hi there @The-ML-Hero!

I think semantic segmentation is already under development as stated in #2.

jrieke commented 3 years ago

Hi @The-ML-Hero that's awesome!

I'm working on object detection with torchvision at the moment but having another template with YOLO would be great! As @themrcesi said, someone else already wanted to have a look at segmentation but I'm not sure if there's any progress yet – let me ask in the other issue.

I've seen your PR for style transfer, is that already finished? Then I'll have a look at it soon (could take a few days as I got some other things going on).