I added new layers to the tiny-VGG and got a new model. When I use CNN-Explainer to lo

Hello <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-ur

detail view problem about cnn-explainer HOT 1 CLOSED

poloclub commented on May 3, 2024

detail view problem

from cnn-explainer.

Comments (1)

xiaohk commented on May 3, 2024

Hello @baobobby, thank you for using CNN Explainer! Glad to see you are trying your own tiny-vgg model!

The CNN Explainer visualization is tailored to the original architecture of tiny-vgg, so you need to manually update the visualization code if you want to add a new layer.

For example, in the Overview.svelte, you would need to update the layerIndexDict definition, and corresponding variables. It is relatively easier to add a new layer that already exists in CNN Explainer than using a totally different model, but it still requires some effort to make the change in the visualization code.

cnn-explainer/src/overview/Overview.svelte

Lines 141 to 154 in b9c1759

    
             const layerIndexDict = { 
        
               'input': 0, 
        
               'conv_1_1': 1, 
        
               'relu_1_1': 2, 
        
               'conv_1_2': 3, 
        
               'relu_1_2': 4, 
        
               'max_pool_1': 5, 
        
               'conv_2_1': 6, 
        
               'relu_2_1': 7, 
        
               'conv_2_2': 8, 
        
               'relu_2_2': 9, 
        
               'max_pool_2': 10, 
        
               'output': 11 
        
             }

cnn-explainer/src/overview/Overview.svelte

Lines 202 to 215 in b9c1759

    
                   let updatingLayerIndexDict = { 
        
                     local: { 
        
                       module: [1, 2, 8, 9, 10], 
        
                       global: [1, 2, 3, 4, 5, 8, 9, 10] 
        
                     }, 
        
                     module: { 
        
                       local: [1, 2, 8, 9, 10], 
        
                       global: [1, 2, 3, 4, 5, 8, 9, 10] 
        
                     }, 
        
                     global: { 
        
                       local: [1, 2, 3, 4, 5, 8, 9, 10], 
        
                       module: [1, 2, 3, 4, 5] 
        
                     } 
        
                   };

I would close the issue for now, let me know if you have other questions :)

from cnn-explainer.

Related Issues (20)

Recommend Projects

detail view problem about cnn-explainer HOT 1 CLOSED

Comments (1)

Related Issues (20)

Recommend Projects

React

Vue.js

Typescript

TensorFlow

Django

Laravel

D3

Recommend Topics

javascript

web

server

Machine learning

Visualization

Game

Recommend Org

Facebook

Microsoft

Google

Alibaba

D3

Tencent

	const layerIndexDict = {
	'input': 0,
	'conv_1_1': 1,
	'relu_1_1': 2,
	'conv_1_2': 3,
	'relu_1_2': 4,
	'max_pool_1': 5,
	'conv_2_1': 6,
	'relu_2_1': 7,
	'conv_2_2': 8,
	'relu_2_2': 9,
	'max_pool_2': 10,
	'output': 11
	}

	let updatingLayerIndexDict = {
	local: {
	module: [1, 2, 8, 9, 10],
	global: [1, 2, 3, 4, 5, 8, 9, 10]
	},
	module: {
	local: [1, 2, 8, 9, 10],
	global: [1, 2, 3, 4, 5, 8, 9, 10]
	},
	global: {
	local: [1, 2, 3, 4, 5, 8, 9, 10],
	module: [1, 2, 3, 4, 5]
	}
	};