[GSoC] Added TF and PyTorch segmentation conversion cases #17801

LupusSanctus · 2020-07-10T13:28:44Z

GSoC project: "Write a tutorial about training a network in TF or PyTorch and then running it in OpenCV"

Part 1. Classification case
Part 2. Segmentation case:

TensorFlow DNN (FCN example) => DNN conversion => OpenCV Net object
PyTorch DNN (FCN example) => DNN conversion => OpenCV Net object
Evaluation

Pull Request Readiness Checklist

See details at https://github.com/opencv/opencv/wiki/How_to_contribute#making-a-good-pull-request

I agree to contribute to the project under OpenCV (BSD) License.
To the best of my knowledge, the proposed patch is not based on a code under GPL or other license that is incompatible with OpenCV
The PR is proposed to proper branch
There is reference to original bug report and related work
There is accuracy test, performance test and test data in opencv_extra repository, if applicable
Patch to opencv_extra has the same branch name.
The feature is well documented and sample code can be built with the project CMake

force_builders_only=Docs

dkurt · 2020-07-20T12:41:41Z

It's not clear for me what this PR proposes. Can you summarize in the description?

asmorkalov · 2020-08-20T10:00:30Z

samples/python/tutorial_code/dnn/dnn_conversion/common/utils.py

+MODEL_PATH_ROOT = os.path.join(DNN_LIB.lower() + "_conversion", "{}/models")
+
+
+def make_dir(dir_to_create):


you can replace the function with os.path.makedirs(dir_to_create, exist_ok=True)

asmorkalov · 2020-08-20T10:02:26Z

samples/python/tutorial_code/dnn/dnn_conversion/common/utils.py

+    plt.title(experiment_name, fontsize=15)
+    plt.legend()
+    plt.savefig(experiment_name.replace(" ", "_") + ".png", bbox_inches="tight")
+    pass


pass is redundant here.

asmorkalov · 2020-08-20T10:07:59Z

samples/python/tutorial_code/dnn/dnn_conversion/common/abstract_model.py

+        pass
+
+
+# https://github.com/opencv/opencv/blob/9ba5581d176baf7c8fceb12b4dcef6c49c7a087d/modules/dnn/test/imagenet_cls_test_alexnet.py


I propose to replace the URL to relative file name in OpenCV repository. The link points to particular commit, but the file changes time to time.

asmorkalov · 2021-01-02T12:43:21Z

@LupusSanctus Friendly reminder.

asmorkalov · 2021-01-26T14:34:17Z

@LupusSanctus the first part has been merged. Please re-base the PR and fix CI-reported issues.

[GSoC] Added TF and PyTorch segmentation conversion cases * WIP: Added conversion of segmentation models * Added tutorial mds * remove unused

ethanc8 · 2024-06-29T15:34:00Z

@LupusSanctus could you please provide the images linked to the tutorials and also the full self-contained segmentation code, especially the implementations of get_colored_mask and get_legend? It's hard to understand your tutorials when there are links to nonexistent images, the referenced sample code file doesn't share any code with the tutorial, and the tutorial doesn't provide the implementations of important functions required to see if the code has been actually working.

vpisarev added the GSoC label Jul 19, 2020

asmorkalov reviewed Aug 20, 2020

View reviewed changes

asmorkalov requested changes Aug 21, 2020

View reviewed changes

LupusSanctus changed the title ~~WIP: Added TF and PyTorch segmentation conversion cases~~ WIP: [GSoC] Added TF and PyTorch segmentation conversion cases Aug 31, 2020

LupusSanctus mentioned this pull request Aug 31, 2020

[GSoC] Added TF and PyTorch detection model conversion cases #18237

Merged

6 tasks

LupusSanctus changed the title ~~WIP: [GSoC] Added TF and PyTorch segmentation conversion cases~~ [GSoC] Added TF and PyTorch segmentation conversion cases Apr 1, 2021

Anastasia added 2 commits April 1, 2021 12:28

WIP: Added conversion of segmentation models

75331b1

Added tutorial mds

b07fdce

LupusSanctus force-pushed the am/pytorch_tf_segm branch from 261f765 to b07fdce Compare April 1, 2021 09:37

remove unused

b170a38

alalek approved these changes Apr 1, 2021

View reviewed changes

alalek merged commit ad333e0 into opencv:master Apr 1, 2021

alalek mentioned this pull request Apr 9, 2021

(5.x) Merge 4.x #19885

Merged

asmorkalov mentioned this pull request Jun 4, 2021

Samples and tutorials for the Dnn High Level API #15240

Closed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Uh oh!

[GSoC] Added TF and PyTorch segmentation conversion cases #17801

[GSoC] Added TF and PyTorch segmentation conversion cases #17801

Uh oh!

LupusSanctus commented Jul 10, 2020 •

edited

Loading

Uh oh!

dkurt commented Jul 20, 2020

Uh oh!

asmorkalov Aug 20, 2020

Uh oh!

asmorkalov Aug 20, 2020

Uh oh!

asmorkalov Aug 20, 2020

Uh oh!

asmorkalov commented Jan 2, 2021

Uh oh!

asmorkalov commented Jan 26, 2021

Uh oh!

ethanc8 commented Jun 29, 2024

Uh oh!

Uh oh!

		MODEL_PATH_ROOT = os.path.join(DNN_LIB.lower() + "_conversion", "{}/models")


		def make_dir(dir_to_create):

		pass


		# https://github.com/opencv/opencv/blob/9ba5581d176baf7c8fceb12b4dcef6c49c7a087d/modules/dnn/test/imagenet_cls_test_alexnet.py

Uh oh!

[GSoC] Added TF and PyTorch segmentation conversion cases #17801

[GSoC] Added TF and PyTorch segmentation conversion cases #17801

Uh oh!

Conversation

LupusSanctus commented Jul 10, 2020 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Pull Request Readiness Checklist

Uh oh!

dkurt commented Jul 20, 2020

Uh oh!

asmorkalov Aug 20, 2020

Choose a reason for hiding this comment

Uh oh!

asmorkalov Aug 20, 2020

Choose a reason for hiding this comment

Uh oh!

asmorkalov Aug 20, 2020

Choose a reason for hiding this comment

Uh oh!

asmorkalov commented Jan 2, 2021

Uh oh!

asmorkalov commented Jan 26, 2021

Uh oh!

ethanc8 commented Jun 29, 2024

Uh oh!

Uh oh!

LupusSanctus commented Jul 10, 2020 •

edited

Loading